tf.size

tf.size

tf.size

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

Defined in tensorflow/python/ops/array_ops.py.

See the guide: Tensor Transformations > Shapes and Shaping

Returns the size of a tensor.

This operation returns an integer representing the number of elements in input.

For example:

登录查看完整内容