size
paddle.fluid.layers.size
(input)[源代码]
返回张量的单元数量,是一个shape为[1]的int64的张量。
参数
- input (Variable)- 输入变量
返回
(Variable)。
代码示例
import paddle.fluid.layers as layers
input = layers.data(
name="input", shape=[3, 100], dtype="float32", append_batch_size=False)
rank = layers.size(input) # 300