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)
>>> print(numel.numpy())
140

         