o Me.@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)v1zlayers.InputSpecc@sFeZdZdZ        d 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|dur t|jnd|_t|}|jdurd}nt|}|dur-t||_ ||_ n||_ d|_ ||_ ||_ ||_||_ zpCifddD|_Wnttfy_tdw|jr|j dusm|j dur|j rs|j n|j d} t|j} | | krtd| | dSdSdS)Ncsi|] }t||qS)int).0kaxesrGD:\Projects\ConvertPro\env\Lib\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_axisrr r __init__KsD     zInputSpec.__init__cCs|jr dt|jnd|jrdt|jnd|jr dt|jnd|jr+dt|jnd|jr6dt|jnd|jrAdt|jndg}dd d d |Dd S) Nzdtype=zshape=zndim=z max_ndim=z min_ndim=zaxes=z InputSpec(z, css|]}|r|VqdSNr)r xrrr sz%InputSpec.__repr__..))rstrrrrrr join)r!specrrr __repr__xszInputSpec.__repr__cCs|j|j|j|j|j|jdS)Nrrrrrr r,)r!rrr get_configszInputSpec.get_configcCs|di|S)Nrr)clsconfigrrr from_configszInputSpec.from_config)NNNNNNFN) __name__ __module__ __qualname____doc__r"r+r- classmethodr0rrrr rs+ -  rcCsd|jdur|jdurtdS|jdurt|jSdg|j}|jD] }|j|||<q#t|S)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      r7cCs|sdStj|}t|tr?dd|D}t|r?g}|D]}||vr5td|dt|d|| ||q|}tj|}|D]}t |dsUt d|qGt |t |krrtd |d t |d t |d |t t||D]P\}\}}|durqyt|j} | jdurdS|jdur|js| j} | |jkrtd |d|d|jd| dt|  |jdur|jj} | dur| |jkrtd |d|d|jd| |jdur|jj} | dur| |jkrtd |d|d|jd| dt|  |jdur'|jj|jkr'td |d|d|jd|j| } |jri|jD]4\} } t | drA| j} | durg| t| | dhvrgtd |d|d| d| dt|j q4|jdur| jdur|j}|jr| r| ddkr| dd} |r|ddkr|dd}t|| D](\}}|dur|dur||krtd |d|d|jdt|jqqydS)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)r r*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 rz1" is incompatible with the layer: expected shape=z, found shape=)rnestflatten isinstancedictallrlistkeysappendhasattrrr enumerateziprrrrrrrrrrrr itemsr9r display_shape) input_specZinputsZ layer_namenamesZ list_inputsrr%Z input_indexr*rrZ shape_as_listZaxisr9Z spec_shapeZspec_dimdimrrr assert_input_compatibilitys               rKcCstt|Sr$)r(rr)rrrr rG/srGcCs<|pt}t|tr|jp|}tt||Std|S)z2Converts a Keras InputSpec object to a TensorSpec.N)rZfloatxr=rrrZ TensorSpecr7)rHZ default_dtyperrrr to_tensor_spec3s    rLr$)r4Ztensorflow.compat.v2compatv2rZkerasrZ tensorflow.python.util.tf_exportrrrr7rKrGrLrrrr s     q