Python tf.keras.layers.InputSpec用法及代码示例

指定层的每个输入的等级、dtype 和形状。

用法

tf.keras.layers.InputSpec(
    dtype=None, shape=None, ndim=None, max_ndim=None, min_ndim=None, axes=None,
    allow_last_axis_squeeze=False, name=None
)

参数

dtype 输入的预期数据类型。
shape 形状元组，输入的预期形状(对于未检查的轴，可能包括无)。包括批量大小。
ndim 整数，输入的预期等级。
max_ndim 整数，输入的最大等级。
min_ndim 整数，输入的最小等级。
axes 将整数轴映射到特定维度值的字典。
allow_last_axis_squeeze 如果为 True，则只要输入的最后一个轴为 1，则允许秩为 N+1 的输入，只要规范的最后一个轴为 1，则允许秩为 N-1 的输入。
name 将数据作为字典传递时与此输入对应的预期键。

层可以公开(如果适用)input_spec 属性：InputSpec 的实例，或 InputSpec 实例的嵌套结构(每个输入张量一个)。这些对象使层能够对输入结构、输入等级、输入形状和输入 dtype 运行输入兼容性检查。

形状中的 None 条目与任何维度兼容， None 形状与任何形状兼容。

例子：

class MyLayer(Layer):
    def __init__(self):
        super(MyLayer, self).__init__()
        # The layer will accept inputs with shape (?, 28, 28) & (?, 28, 28, 1)
        # and raise an appropriate error message otherwise.
        self.input_spec = InputSpec(
            shape=(None, 28, 28, 1),
            allow_last_axis_squeeze=True)

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.layers.InputSpec。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。