rank

paddle. rank ( input ) [源代码]

计算输入 Tensor 的维度(秩)。

参数

  • input (Tensor) — 输入 input 是 shape 为 \([N_1, N_2, ..., N_k]\) 的多维 Tensor,数据类型可以任意类型。

返回

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

代码示例

import paddle

input = paddle.rand((3, 100, 100))
rank = paddle.rank(input)
print(rank)
# 3