B d.@sdZddlmmZddlmZddlmZddlm Z edddgde d gdGd d d Z d d Z ddZ ddZ dddZdS)zContains the InputSpec class.N)backend) keras_export) tf_exportzkeras.layers.InputSpecz*keras.__internal__.legacy.layers.InputSpec)Zv1zlayers.InputSpecc@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)axesrI/var/www/html/venv/lib/python3.7/site-packages/keras/engine/input_spec.py hsz&InputSpec.__init__..zAArgument `axes` must be a dict with integer keys. Received: axes=z5Axis {} is greater than the maximum allowed value: {})tfZas_dtypenamedtype TensorShaperanktupleas_listlenndimshapemax_ndimmin_ndimallow_last_axis_squeezer ValueError TypeErrormaxformat) selfrrrrrr rrZmax_dimZmax_axisr)r r __init__Ks6       zInputSpec.__init__cCs|jrdt|jnd|jr*dt|jnd|jr@dt|jnd|jrVdt|jnd|jrldt|jnd|jrdt|jndg}dd d d |Dd S) Nzdtype=zshape=zndim=z max_ndim=z min_ndim=zaxes=z InputSpec(z, css|]}|r|VqdS)Nr)rxrrr sz%InputSpec.__repr__..))rstrrrrrr join)rspecrrr __repr__xszInputSpec.__repr__cCs|j|j|j|j|j|jdS)N)rrrrrr )rrrrrr )rrrr get_configs zInputSpec.get_configcCs |f|S)Nr)clsconfigrrr from_configszInputSpec.from_config)NNNNNNFN) __name__ __module__ __qualname____doc__r r(r) classmethodr,rrrr rs- $  rcCsl|jdkr|jdkrtdS|jdk r4t|jSdg|j}x|jD]}|j|||<qHWt|SdS)a@Returns 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'rarrr to_tensor_shapes      r3cCs|sdStj|}t|trdd|D}t|rg}xD|D]<}||krltd|dt|d|| ||q>W|}tj|}x$|D]}t |dst d|qWt |t |krtd |d t |d t |d |xt t||D]\}\}}|dkrqt|j} | jdkr4dS|jdk r|js| j} | |jkrtd |d|d|jd| dt|  |jdk r|jj} | dk r| |jkrtd |d|d|jd| |jdk r(|jj} | dk r(| |jkr(td |d|d|jd| dt|  |jdk rh|jj|jkrhtd |d|d|jd|j| } |jrxv|jD]h\} } t | dr| j} | dk r| t| | dhkrtd |d|d| d| dt|j qW|jdk r| jdk r|j}|jrR| r2| ddkr2| dd} |rR|ddkrR|dd}x^t|| D]P\}}|dk r^|dk r^||kr^td |d|d|jdt|jq^WqWdS)aChecks compatibility between the layer and provided inputs. This checks that the tensor(s) `inputs` verify the input assumptions of a layer (if any). If not, a clear and actional exception gets raised. Args: input_spec: An InputSpec instance, list of InputSpec instances, a nested structure of InputSpec instances, or None. inputs: Input tensor, list of input tensors, or a nested structure of input tensors. layer_name: String, name of the layer (for error message formatting). Raises: ValueError: in case of mismatch between the provided inputs and the expectations of the layer. NcSsg|] }|jqSr)r)rr'rrr sz.assert_input_compatibility..zMissing data for input "z*". You passed a data dictionary with keys z. Expected the following keys: rz*Inputs to a layer should be tensors. Got: zLayer "z " expects z input(s), but it received z! input tensors. Inputs received: zInput z of layer "z0" is incompatible with the layer: expected ndim=z , found ndim=z. Full shape received: z4" is incompatible with the layer: expected max_ndim=z4" is incompatible with the layer: expected min_ndim=z1" is incompatible with the layer: expected dtype=z, found dtype=valuez0" is incompatible with the layer: expected axis z of input shape to have value z , but received input with shape r z1" is incompatible with the layer: expected shape=z, found shape=)rnestflatten isinstancedictallrlistkeysappendhasattrrr enumerateziprrrrrrrrrrrr itemsr5r display_shape) input_specinputsZ layer_namenamesZ list_inputsrr"Z input_indexr'rrZ shape_as_listZaxisr5Z spec_shapeZspec_dimdimrrr assert_input_compatibilitys       &     *   * "  0   rHcCstt|S)N)r%rr)rrrr rC/srCcCs<|p t}t|tr0|jp|}tt||Std|S)z2Converts a Keras InputSpec object to a TensorSpec.N)rZfloatxr9rrrZ TensorSpecr3)rDZ default_dtyperrrr to_tensor_spec3s    rI)N)r0Ztensorflow.compat.v2compatZv2rZkerasrZ tensorflow.python.util.tf_exportrrrr3rHrCrIrrrr s    r