numel

paddle. numel ( x, name=None )

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

参数

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

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

返回

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

代码示例

COPY-FROM: paddle.numel