InputSpec¶
- class paddle.static. InputSpec ( shape, dtype='float32', name=None, stop_gradient=False ) ¶
用于描述模型输入的签名信息，包括 shape、dtype 和 name。
此接口常用于指定高层 API 中模型的输入 Tensor 信息，或动态图转静态图时，指定被 paddle.jit.to_static 装饰的 forward 函数每个输入参数的 Tensor 信息。
参数¶
shape (tuple(integers)|list[integers])- 声明维度信息的 list 或 tuple，默认值为 None。设置为
None或-1时表示维度可以是任意大小。例如，可以将可变的批尺寸（batch size）设置为None或-1。dtype (np.dtype|str，可选)- 数据类型，支持 bool，float16，float32，float64，int8，int16，int32，int64，uint8。默认值为 float32。
name (str，可选) - 变量的名称，具体用法请参见 Name，一般无需设置，默认值为 None。
stop_gradient (bool，可选) - 提示是否应该停止计算梯度，默认值为 False，表示不停止计算梯度。
返回¶
初始化后的 InputSpec 对象。
代码示例¶
COPY-FROM: paddle.static.InputSpec
方法¶
from_tensor(tensor, name=None)¶
该接口将根据输入 Tensor 的 shape、dtype 等信息构建 InputSpec 对象。
参数
tensor (Tensor) - 用于构建 InputSpec 的源 Tensor
name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。
返回
根据 Tensor 信息构造的 InputSpec 对象。
代码示例
COPY-FROM: paddle.static.InputSpec.from_tensor
from_numpy(ndarray, name=None)¶
该接口将根据输入 numpy ndarray 的 shape、dtype 等信息构建 InputSpec 对象。
参数
ndarray (Tensor) - 用于构建 InputSpec 的 numpy ndarray
name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。
返回
根据 ndarray 信息构造的 InputSpec 对象。
代码示例
COPY-FROM: paddle.static.InputSpec.from_numpy
batch(batch_size)¶
该接口将 batch_size 插入到当前 InputSpec 对象的 shape 元组最前面。
参数
batch_size (int) - 被插入的 batch size 整型数值
返回
更新 shape 信息后的
InputSpec对象。
代码示例
COPY-FROM: paddle.static.InputSpec.batch
unbatch()¶
该接口将当前 InputSpec 对象 shape[0]值移除。
返回
更新 shape 信息后的
InputSpec对象。
代码示例
COPY-FROM: paddle.static.InputSpec.unbatch