o ?e+@sdZddlmZddlmZddlmZddlmZddlmZddl m Z e dgd Gd d d e Z d d Z ddZddZdddZdS)zContains the InputSpec class.)dtypes) tensor_shape) tensor_spec)backend)nest) tf_exportzlayers.InputSpec)v1c@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|_Wn ttfy\tdw|jr|j dusj|j dur|j rp|j n|j d} t|j} | | krtd| | dSdSdS)Ncsi|] }t||qS)int).0kaxesr j/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/keras/engine/input_spec.py `sz&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__Es:     zInputSpec.__init__cCs|jr dt|jnd|jrdt|jnd|jr dt|jnd|jr+dt|jnd|jr6dt|jnd|jrAdt|jndg}dd d d |DS) Nzdtype=zshape=zndim=z max_ndim=z min_ndim=zaxes=z InputSpec(%s)z, css|]}|r|VqdSNr )r xr r r rsz%InputSpec.__repr__..)rstrrrrrrjoin)r#specr r r__repr__kszInputSpec.__repr__cCs|j|j|j|j|j|jdS)Nrrrrrrr-)r#r r r get_configtszInputSpec.get_configcCs|di|S)Nr r )clsconfigr r r from_config}szInputSpec.from_config)NNNNNNFN) __name__ __module__ __qualname____doc__r$r,r. classmethodr1r r r rr s) &  r cCsd|jdur|jdurtdS|jdurt|jSdg|j}|jD] }|j|||<q#t|S)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      r8cCs|sdSt|}t|tr:dd|D}t|r:g}|D]}||vr0td|t||f|||q|}t|}|D]}t |dsOt d|fqAt |t |krttd|dt t |d t t |d t |t t||D]\}\}}|durq{t|j} | jdurdS|jdur|js| j} | |jkrtd t |d |d t |jdt | dt t| |jdur|jj} | dur| |jkrtd t |d |dt |jdt | |jdur'|jj} | dur'| |jkr'td t |d |dt |jdt | dt t| |jdurO|jj|jkrOtd t |d |dt |jdt |j| } |jr|jD]=\} } t | dri| j} | dur| t| | dhvrtd t |d |dt | dt | dt|jq\|jdur| jdur|j}|jr| r| ddkr| dd} |r|ddkr|dd}t|| D].\}}|dur|dur||krtd t |d|dt |jdt|jqq{dS)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+r r r sz.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 isinstancedictallrlistkeysappendhasattrr rr) enumerateziprrrrrrrrrrrrritemsr:r display_shape) input_specZinputsZ layer_namenamesZ list_inputsrr'Z input_indexr+rrZ shape_as_listZaxisr:Z spec_shapeZspec_dimdimr r rassert_input_compatibilitys"                     rKcCstt|Sr&)r)rr)rr r rrGsrGNcCs<|pt}t|tr|jp|}tt||Std|S)z2Converts a Keras InputSpec object to a TensorSpec.N)rZfloatxr=r rrZ TensorSpecr8)rHZ default_dtyperr r rto_tensor_specs    rLr&)r5Ztensorflow.python.frameworkrrrZtensorflow.python.kerasrZtensorflow.python.utilrZ tensorflow.python.util.tf_exportrobjectr r8rKrGrLr r r rs       fv