size

paddle.fluid.layers.size(input)[源代码]

返回张量的单元数量,是一个shape为[1]的int64的张量。

参数

  • input (Variable)- 输入变量

返回

(Variable)。

代码示例

  1. import paddle.fluid.layers as layers
  2. input = layers.data(
  3. name="input", shape=[3, 100], dtype="float32", append_batch_size=False)
  4. rank = layers.size(input) # 300