numel

paddle. numel ( x ) [源代码]

在静态图模式下,返回一个长度为 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