rank
paddle.fluid.layers.
rank
(input)[源代码]
该OP用于计算输入Tensor的维度(秩)。
- 参数:
- input (Variable) — 输入input是shape为 的多维Tensor,数据类型可以任意类型。
返回:输出Tensor的秩,是一个0-D Tensor。
返回类型:Variable,数据类型为int32。
代码示例
- import paddle.fluid as fluid
- input = fluid.data(
- name="input", shape=[3, 100, 100], dtype="float32")
- rank = fluid.layers.rank(input) # rank=(4,)