rank

paddle. rank ( input ) [源代码]

该OP用于计算输入Tensor的维度(秩)。

参数:

  • input (Tensor) — 输入input是shape为

    rank - 图1

    的多维Tensor,数据类型可以任意类型。

返回:输出Tensor的秩,是一个0-D Tensor。

代码示例

  1. import paddle
  2. input = paddle.rand((3, 100, 100))
  3. rank = paddle.rank(input)