o 2ht@sZddlZddlmZddlmZddlmZedGdddeZedd d d ZdS) N)ops) keras_export)Mergezkeras.layers.ConcatenatecsPeZdZdZdfdd ZddZddZd d Zdd d ZfddZ Z S) ConcatenateaConcatenates a list of inputs. It takes as input a list of tensors, all of the same shape except for the concatenation axis, and returns a single tensor that is the concatenation of all inputs. Examples: >>> x = np.arange(20).reshape(2, 2, 5) >>> y = np.arange(20, 30).reshape(2, 1, 5) >>> keras.layers.Concatenate(axis=1)([x, y]) Usage in a Keras model: >>> x1 = keras.layers.Dense(8)(np.arange(10).reshape(5, 2)) >>> x2 = keras.layers.Dense(8)(np.arange(10, 20).reshape(5, 2)) >>> y = keras.layers.Concatenate()([x1, x2]) Args: axis: Axis along which to concatenate. **kwargs: Standard layer keyword arguments. Returns: A tensor, the concatenation of the inputs alongside axis `axis`. c s(tjdi|||_d|_d|_dS)NTF)super__init__axissupports_masking_reshape_required)selfr kwargs __class__r_/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/layers/merging/concatenate.pyr $s zConcatenate.__init__c sft|dkst|dttfstd|tdd|Dr!dSdd|D}t|}t}tt|D];}|j t||}t |ddD]\}|krX|dkrX||=qGt|||j krh|||j =| t||q6t|dkrd |}td d|D} t| dkrt|| \} t| D]tfd d|D} t| dkrt|qdSdS) Nrz\A `Concatenate` layer should be called on a list of at least 1 input. Received: input_shape=cs|]}|duVqdSNr.0shaperrr 3z$Concatenate.build..cSsg|]}t|qSr)listrrrr 6sz%Concatenate.build..)startztA `Concatenate` layer requires inputs with matching shapes except for the concatenation axis. Received: input_shape=css|]}t|VqdSr)lenrrrrrUrc3s$|] }|dur|VqdSrrrr rrr]s ) r isinstancetupler ValueErrorallcopysetranger enumerateadd) r input_shapereduced_inputs_shapesreduced_inputs_shapes_copy shape_seti concat_axis axis_valueerr_msgranksrank unique_dimsrrrbuild*sT          zConcatenate.buildcCstj||jdS)Nr)r concatenater )r inputsrrr_merge_functioneszConcatenate._merge_functioncCst|ttfrt|dttfstd||}t|d}|ddD]&}||jdus5||jdur?d||j<t|S||j||j7<q%t|S)NrzRA `Concatenate` layer should be called on a list of inputs. Received: input_shape=r)rr rr!r )r r( input_shapes output_shaperrrrcompute_output_shapehs"  z Concatenate.compute_output_shapeNc Cs"|durdSt|ttfstd|t|ttfs"td|t|t|kr>td|dt|d|dt|tdd|DrIdSg}t||D]0\}}|durc|tj |dd qP|j |j kr{|t tj |d d t |qP||qPtj||jd }tj|d d d S)Nz'`mask` should be a list. Received mask=z,`inputs` should be a list. Received: inputs=zLThe lists `inputs` and `mask` should have the same length. Received: inputs=z of length z , and mask=csrrr)rmrrrrrz+Concatenate.compute_mask..bool)dtyperrF)r keepdims)rr rr!rr"zipappendr ones_likendim broadcast_to expand_dimsrr4r any)r r5maskmasksinput_imask_i concatenatedrrr compute_maskzsF  zConcatenate.compute_maskcs0d|ji}t}tt|t|S)Nr )r r get_configdictritems)r config base_configrrrrKs  zConcatenate.get_configrr) __name__ __module__ __qualname____doc__r r3r6r9rJrK __classcell__rrrrrs; %rzkeras.layers.concatenatercKstdd|i||S)aFunctional interface to the `Concatenate` layer. Args: inputs: A list of input tensors. axis: Concatenation axis. **kwargs: Standard layer keyword arguments. Returns: A tensor, the concatenation of the inputs alongside axis `axis`. r Nr)r)r5r rrrrr4s r4rP) r# keras.srcrkeras.src.api_exportr#keras.src.layers.merging.base_mergerrr4rrrrs