TensorFlow函数:tf.layers.InputSpec
2018-07-12 13:58 更新
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
)
初始化自我.