rank

paddle.rank ( input ) [源代码]

该OP用于计算输入Tensor的维度（秩）。

参数：

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

  

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

返回：输出Tensor的秩，是一个0-D Tensor。

代码示例

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