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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。