a Sic.@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@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 |durt|jnd|_t|}|jdur4d}n t|}|durZt||_ ||_ n ||_ d|_ ||_ ||_ ||_||_ z pifddD|_Wn$ttfytdYn0|jr|j dus|j dur|j r|j n|j d} t|j} | | krtd| | dS)Ncsi|]}t||qS)int).0kaxesrS/var/www/html/django/DPS/env/lib/python3.9/site-packages/keras/engine/input_spec.py hz&InputSpec.__init__..zAArgument `axes` must be a dict with integer keys. Received: axes=z5Axis {} is greater than the maximum allowed value: {})tfas_dtypenamedtype TensorShaperanktupleas_listlenndimshapemax_ndimmin_ndimallow_last_axis_squeezer ValueError TypeErrormaxformat) selfrrrrrr rrmax_dimmax_axisrr r __init__Ks@        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|VqdSNr)r xrrr rz%InputSpec.__repr__..)rstrrrrrr join)r#specrrr __repr__xszInputSpec.__repr__cCs|j|j|j|j|j|jdS)Nrrrrrr r/)r#rrr get_configszInputSpec.get_configcCs|fi|Sr(r)clsconfigrrr from_configszInputSpec.from_config)NNNNNNFN) __name__ __module__ __qualname____doc__r&r.r0 classmethodr3rrrr rs+ -  rcCsh|jdur|jdurtdS|jdur4t|jSdg|j}|jD]}|j|||<qFt|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      r:cCs|sdStj|}t|tr~dd|D}t|r~g}|D]<}||vrjtd|dt|d|| ||q<|}tj|}|D]}t |dst d|qt |t |krtd |d t |d t |d |t t||D]\}\}}|dur qt|j} | jdur*dS|jdur||js|| j} | |jkr|td |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| } |jr|jD]h\} } t | dr| j} | durx| t| | dhvrxtd |d|d| d| dt|j qx|jdur| jdur|j}|jrD| r$| ddkr$| dd} |rD|ddkrD|dd}t|| D]P\}}|durN|durN||krNtd |d|d|jdt|jqNqdS)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 rz.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 isinstancedictallrlistkeysappendhasattrr r enumerateziprrrrrrrrrrrr itemsr<r display_shape) input_specinputs layer_namenames list_inputsrr) input_indexr-rr shape_as_listaxisr< spec_shapespec_dimdimrrr assert_input_compatibilitys                rVcCstt|Sr()r+rr)rrrr rJ/srJcCs<|p t}t|tr0|jp|}tt||Std|S)z2Converts a Keras InputSpec object to a TensorSpec.N)rfloatxr@rrr TensorSpecr:)rK default_dtyperrrr to_tensor_spec3s    rZ)N)r7tensorflow.compat.v2compatv2rkerasr tensorflow.python.util.tf_exportrrrr:rVrJrZrrrr s    q