numel

paddle. numel ( x ) [源代码]

返回 shape 为[]的 0-D Tensor,其值为输入 Tensor 中元素的个数。

参数

  • x (Tensor) - 输入 Tensor,数据类型为 int32、int64、float16、float32、float64、int32、int64。

  • name (str, 可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

返回值为 x 元素个数的 0-D Tensor。

代码示例

import paddle

x = paddle.full(shape=[4, 5, 7], fill_value=0, dtype='int32')
numel = paddle.numel(x) # 140