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