o 2hy&@sJddlmZddlmZddlmZeddgGdddZdd Zd S) )backend)tree) keras_exportzkeras.InputSpeczkeras.layers.InputSpecc@sHeZdZdZ         d ddZddZdd Zed d ZdS) InputSpecaSpecifies 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 for the first argument of `Layer.__call__`. A `None` entry in a shape is compatible with any dimension. Args: dtype: Expected dtype of the input. shape: Shape tuple, expected shape of the input (may include `None` for dynamic 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`, 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. optional: Boolean, whether the input is optional or not. An optional input can accept `None` values. Example: ```python class MyLayer(Layer): def __init__(self): super().__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|nd|_|durt||_t||_n||_d|_||_||_||_ | |_ ||_ zp5ifddD|_ Wnt tfyQtdw|j r{|jdus_|jdur}|jre|jn|jd} t|j } | | krt d| | dSdSdS)Ncsi|] }t||qS)int).0kaxesrV/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/layers/input_spec.py Osz&InputSpec.__init__..zAArgument `axes` must be a dict with integer keys. Received: axes=z5Axis {} is greater than the maximum allowed value: {})rstandardize_dtypedtypestandardize_shapeshapelenndimmax_ndimmin_ndimnameoptionalallow_last_axis_squeezer ValueError TypeErrormaxformat) selfrrrrrr rrrmax_dimmax_axisrr r __init__3s@    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|VqdS)Nr)rxrrr hsz%InputSpec.__repr__..))rstrrrrrr join)rspecrrr __repr___szInputSpec.__repr__cCs|j|j|j|j|j|jdS)Nrrrrrr r*)rrrr get_configjszInputSpec.get_configcCs|di|S)Nrr)clsconfigrrr from_configtszInputSpec.from_config) NNNNNNFNF) __name__ __module__ __qualname____doc__r!r)r+ classmethodr.rrrr rs - ,  rc CsR|sdSt|}t|tr>dd|D}t|r>g}|D]}||vr4td|dt|d||||q|}t|}t |t |kr_td|dt |d t |d |t t ||D]?\}\}}|durrqf|durz|j rzqft |d std |d t|d|dt|j} t | } |jdur|js| |jkrtd|d|d|jd| d| |jdur| dur| |jkrtd|d|d|jd| |jdur| dur| |jkrtd|d|d|jd| d| |jdurt|j} | |jkrtd|d|d|jd| |jrN|jD]&\} } | durL| | | dhvrLtd|d|d| d| d| q'|jdur|j}|jr{| rk| ddkrk| dd} |r{|ddkr{|dd}t || D]%\}}|dur|dur||krtd|d|d|jd| qqfdS) 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: zLayer "z " expects z input(s), but it received z! input tensors. Inputs received: rz*Inputs to a layer should be tensors. Got 'z ' (of type z) as input for layer 'z'.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=z0" 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=)rflatten isinstancedictallrlistkeysappendr enumerateziprhasattrtyperrrrrrrrrr items) input_specinputs layer_namenames list_inputsr input_indexr#r(rrraxisvalue spec_shapespec_dimdimrrr assert_input_compatibilityys                rMN) keras.srcrrkeras.src.api_exportrrrMrrrr s    r