TensorFlow函数:tf.layers.InputSpec

由 Carrie 创建, 最后一次修改 2018-07-12

tf.layers.InputSpec函数

InputSpec类

别名:

  • 类 tf.keras.layers.InputSpec
  • 类 tf.layers.InputSpec

定义在:tensorflow/python/layers/base.py.

指定图层的每个输入的ndim,dtype和形状.

每个层都应公开(如果适用)一个input_spec属性:InputSpec的实例列表(每个输入张量一个).

形状中的“None”条目与任何维度兼容,None形状与任何形状兼容.

参数:

  • dtype:输入的预期DataType.
  • shape:形状元组,输入的预期形状(未经检查的轴可能包括无).
  • ndim:整数,输入的预期秩.
  • max_ndim:整数,输入的最大秩.
  • min_ndim:整数,输入的最小秩.
  • axes:字典将整数轴映射到特定的维值.

方法

__init__

__init__(
    dtype=None,
    shape=None,
    ndim=None,
    max_ndim=None,
    min_ndim=None,
    axes=None
)

初始化自我.

以上内容是否对您有帮助:

二维码
建议反馈
二维码