PaddlePaddle

API 指南

 paddle / rank

rank

paddle. rank ( input ) [源代码]

该OP用于计算输入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)

此页内容是否对您有帮助

上一篇 下一篇
©Copyright 2011-2021 www.yanxishe.com ©AI研习社 All Rights Reserved 粤ICP备11095991号-21