TensorFlow張量變換：tf.size

tf.size 函數(shù)

size(
    input,
    name=None,
    out_type=tf.int32
)

定義在：tensorflow/python/ops/array_ops.py.

參見指南：張量變換>形狀的確定與改變

返回張量的大小.
此操作返回一個(gè)表示input中的元素?cái)?shù)的整數(shù).

例如：

t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
tf.size(t)  # 12

函數(shù)參數(shù)

• input：一個(gè)Tensor或SparseTensor.
• name：操作的名稱(可選).
• out_type:(可選)操作的指定輸出類型(int32或int64)；默認(rèn)為tf.int32.

函數(shù)返回

tf.size函數(shù)返回out_type中類型的Tensor；默認(rèn)為tf.int32.