size¶

paddle.fluid.layers.nn. size ( input ) [source]

Size Layer

Returns the number of elements for a tensor, which is a int64 Tensor with shape [1].

Parameters

input (Tensor) – The input Tensor, it’s data type can be bool, float16, float32, float64, int32, int64.

Returns

The number of elements for the input Tensor.

Return type

Tensor

Raises

TypeError – input must be a Tensor and the data type of input must be one of bool, float16, float32, float64, int32, int64.

Examples

import paddle
import paddle.fluid.layers as layers
paddle.enable_static()

input = layers.data(
    name="input", shape=[3, 100], dtype="float32", append_batch_size=False)
rank = layers.size(input) # 300