B ӻd+@sdZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z e d e d gd Gd d d e Z ddZddZddZdddZdS)zContains the InputSpec class.)dtypes) tensor_shape) tensor_spec)backend)nest) keras_export) tf_exportzkeras.layers.InputSpeczlayers.InputSpec)Zv1c@s6eZdZdZd ddZddZdd Zed d ZdS) InputSpeca Specifies the rank, dtype and shape of every input to a layer. Layers can expose (if appropriate) an `input_spec` attribute: an instance of `InputSpec`, or a nested structure of `InputSpec` instances (one per input tensor). These objects enable the layer to run input compatibility checks for input structure, input rank, input shape, and input dtype. A None entry in a shape is compatible with any dimension, a None shape is compatible with any shape. Args: dtype: Expected DataType of the input. shape: Shape tuple, expected shape of the input (may include None for unchecked axes). Includes the batch size. ndim: Integer, expected rank of the input. max_ndim: Integer, maximum rank of the input. min_ndim: Integer, minimum rank of the input. axes: Dictionary mapping integer axes to a specific dimension value. allow_last_axis_squeeze: If True, then allow inputs of rank N+1 as long as the last axis of the input is 1, as well as inputs of rank N-1 as long as the last axis of the spec is 1. name: Expected key corresponding to this input when passing data as a dictionary. Example: ```python 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) ``` NFc s|dk rt|jnd|_t|}|jdkr4d}n t|}|dk rZt ||_ ||_ n ||_ d|_ ||_ ||_ ||_||_y pifddD|_Wn ttfk rtdYnX|jr|j dk s|j dk r|j r|j n|j d} t|j} | | krtd| | dS)Ncsi|]}|t|qS)int).0k)axesr [/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/keras/engine/input_spec.py bsz&InputSpec.__init__..z"The keys in axes must be integers.z5Axis {} is greater than the maximum allowed value: {})rZas_dtypenamedtyper TensorShaperanktupleas_listlenndimshapemax_ndimmin_ndimallow_last_axis_squeezer ValueError TypeErrormaxformat) selfrrrrrrrrZmax_dimZmax_axisr )rr__init__Gs2       zInputSpec.__init__cCs|jrdt|jnd|jr*dt|jnd|jr@dt|jnd|jrVdt|jnd|jrldt|jnd|jrdt|jndg}dd d d |DS) Nzdtype=zshape=zndim=z max_ndim=z min_ndim=zaxes=z InputSpec(%s)z, css|]}|r|VqdS)Nr )r xr r r tsz%InputSpec.__repr__..)rstrrrrrrjoin)r"specr r r__repr__mszInputSpec.__repr__cCs|j|j|j|j|j|jdS)N)rrrrrr)rrrrrr)r"r r r get_configvs zInputSpec.get_configcCs |f|S)Nr )clsconfigr r r from_configszInputSpec.from_config)NNNNNNFN) __name__ __module__ __qualname____doc__r#r*r+ classmethodr.r r r rr s)   r cCsl|jdkr|jdkrtdS|jdk r4t|jSdg|j}x|jD]}|j|||<qHWt|SdS)a2Returns a tf.TensorShape object that matches the shape specifications. If the InputSpec's shape or ndim is defined, this method will return a fully or partially-known shape. Otherwise, the returned TensorShape is None. Args: spec: an InputSpec object. Returns: a tf.TensorShape object N)rrrrr)r)rar r rto_tensor_shapes      r5cCs0|sdSt|}t|trxdd|D}t|rxg}x<|D]4}||krbtd|t||f|||qsz.assert_input_compatibility..zgMissing data for input "%s". You passed a data dictionary with keys %s. Expected the following keys: %srz,Inputs to a layer should be tensors. Got: %szLayer z expects z input(s), but it received z! input tensors. Inputs received: zInput z of layer z/ is incompatible with the layer: expected ndim=z , found ndim=z. Full shape received: z3 is incompatible with the layer: expected max_ndim=z5 is incompatible with the layer: : expected min_ndim=z0 is incompatible with the layer: expected dtype=z, found dtype=valuez/ is incompatible with the layer: expected axis z of input shape to have value z but received input with shape rz is incompatible with layer z: expected shape=z, found shape=) rflatten isinstancedictallrlistkeysappendhasattrrrr' enumerateziprrrrrrrrrrrrritemsr7r display_shape) input_specinputsZ layer_namenamesZ list_inputsrr%Z input_indexr)rrZ shape_as_listZaxisr7Z spec_shapeZspec_dimdimr r rassert_input_compatibilitys      6     0 0 0 2  B   rIcCstt|S)N)r'rr)rr r rrDsrDNcCs<|p t}t|tr0|jp|}tt||Std|S)z2Converts a Keras InputSpec object to a TensorSpec.N)rZfloatxr:r rrZ TensorSpecr5)rEZ default_dtyperr r rto_tensor_specs    rJ)N)r2Ztensorflow.python.frameworkrrrZtensorflow.python.kerasrZtensorflow.python.utilrZ tensorflow.python.util.tf_exportrrobjectr r5rIrDrJr r r rs        fv