numel¶
在静态图模式下,返回一个长度为 1 并且元素值为输入 x
元素个数的 Tensor;在动态图模式下,返回一个标量数值。
参数¶
x (Tensor) - 输入 Tensor,数据类型为 int32、int64、float16、float32、float64、int32、int64。
name (str, 可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
返回长度为 1 并且元素值为 x
元素个数的 Tensor。
代码示例¶
import paddle
x = paddle.full(shape=[4, 5, 7], fill_value=0, dtype='int32')
numel = paddle.numel(x) # 140