paddle.fluid / layers / size

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

此页内容是否对您有帮助

PaddlePaddle

paddle

paddle.amp

paddle.callbacks

paddle.compat

paddle.distributed

paddle.distribution

paddle.fluid

paddle.io

paddle.jit

paddle.metric

paddle.nn

paddle.onnx

paddle.optimizer

paddle.regularizer

paddle.static

paddle.sysconfig

paddle.text

paddle.utils

paddle.vision

paddle.fluid / layers / size

size¶