size¶

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

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

参数¶

input （Variable）- 输入变量

返回¶

(Variable)。

代码示例¶

import paddle
import paddle.fluid.layers as layers
paddle.enable_static()

input = layers.data(
    name="input", shape=[3, 100], dtype="float32", append_batch_size=False)
rank = layers.size(input) # 300