numel
paddle. numel ( x ) [源代码]
该OP返回一个长度为1并且元素值为输入 x
元素个数的Tensor。
参数:
- x (Tensor) - 输入Tensor,数据类型为int32,int64, float16, float32, float64, int32, int64 。
返回: 返回长度为1并且元素值为 x
元素个数的Tensor。
代码示例:
import paddle
x = paddle.full(shape=[4, 5, 7], fill_value=0, dtype='int32')
numel = paddle.numel(x) # 140