TensorFlow张量变换:tf.size

由 Carrie 创建, 最后一次修改 2018-01-25

tf.size 函数

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

定义在:tensorflow/python/ops/array_ops.py.

参见指南:张量变换>形状的确定与改变

返回张量的大小.
此操作返回一个表示input中的元素数的整数.

例如:

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

函数参数

  • input:一个Tensor或SparseTensor.
  • name:操作的名称(可选).
  • out_type:(可选)操作的指定输出类型(int32或int64);默认为tf.int32.

函数返回

tf.size函数返回out_type中类型的Tensor;默认为tf.int32.

以上内容是否对您有帮助:

二维码
建议反馈
二维码