
    6bi"                     v    S r SSKJs  Jr  SSKJr  SSKJr  SSKJ	r	  SSK
Jr  \" S5       " S S	\5      5       rg)
zKeras depthwise 1D convolution.    N)DepthwiseConv)
conv_utils)tf_utils)keras_exportzkeras.layers.DepthwiseConv1Dc                   v   ^  \ rS rSrSr              SU 4S jjrS r\R                  S 5       r	Sr
U =r$ )DepthwiseConv1D   a  Depthwise 1D convolution.

Depthwise convolution is a type of convolution in which each input channel
is convolved with a different kernel (called a depthwise kernel). You can
understand depthwise convolution as the first step in a depthwise separable
convolution.

It is implemented via the following steps:

- Split the input into individual channels.
- Convolve each channel with an individual depthwise kernel with
  `depth_multiplier` output channels.
- Concatenate the convolved outputs along the channels axis.

Unlike a regular 1D convolution, depthwise convolution does not mix
information across different input channels.

The `depth_multiplier` argument determines how many filter are applied to
one input channel. As such, it controls the amount of output channels that
are generated per input channel in the depthwise step.

Args:
  kernel_size: An integer, specifying the height and width of the 1D
    convolution window. Can be a single integer to specify the same value
    for all spatial dimensions.
  strides: An integer, specifying the strides of the convolution along the
    height and width. Can be a single integer to specify the same value for
    all spatial dimensions. Specifying any stride value != 1 is incompatible
    with specifying any `dilation_rate` value != 1.
  padding: one of `'valid'` or `'same'` (case-insensitive). `"valid"` means
    no padding. `"same"` results in padding with zeros evenly to the
    left/right or up/down of the input such that output has the same
    height/width dimension as the input.
  depth_multiplier: The number of depthwise convolution output channels for
    each input channel. The total number of depthwise convolution output
    channels will be equal to `filters_in * depth_multiplier`.
  data_format: A string, one of `channels_last` (default) or
    `channels_first`.  The ordering of the dimensions in the inputs.
    `channels_last` corresponds to inputs with shape `(batch_size, height,
    width, channels)` while `channels_first` corresponds to inputs with
    shape `(batch_size, channels, height, width)`. When unspecified, uses
    `image_data_format` value found in your TF-Keras config file at
    `~/.keras/keras.json` (if exists) else 'channels_last'.
    Defaults to 'channels_last'.
  dilation_rate: A single integer, specifying the dilation rate to use for
    dilated convolution. Currently, specifying any `dilation_rate`
    value != 1 is incompatible with specifying any stride value != 1.
  activation: Activation function to use. If you don't specify anything, no
    activation is applied (see `keras.activations`).
  use_bias: Boolean, whether the layer uses a bias vector.
  depthwise_initializer: Initializer for the depthwise kernel matrix (see
    `keras.initializers`). If None, the default initializer
    ('glorot_uniform') will be used.
  bias_initializer: Initializer for the bias vector (see
    `keras.initializers`). If None, the default initializer ('zeros') will
    be used.
  depthwise_regularizer: Regularizer function applied to the depthwise
    kernel matrix (see `keras.regularizers`).
  bias_regularizer: Regularizer function applied to the bias vector (see
    `keras.regularizers`).
  activity_regularizer: Regularizer function applied to the output of the
    layer (its 'activation') (see `keras.regularizers`).
  depthwise_constraint: Constraint function applied to the depthwise kernel
    matrix (see `keras.constraints`).
  bias_constraint: Constraint function applied to the bias vector (see
    `keras.constraints`).

Input shape:
  3D tensor with shape: `[batch_size, channels, input_dim]` if
    data_format='channels_first'
  or 3D tensor with shape: `[batch_size, input_dim, channels]` if
    data_format='channels_last'.

Output shape:
  3D tensor with shape:
   `[batch_size, channels * depth_multiplier, new_dims]`
    if `data_format='channels_first'`
    or 3D tensor with shape: `[batch_size,
    new_dims, channels * depth_multiplier]` if
    `data_format='channels_last'`. `new_dims` values might have
    changed due to padding.

Returns:
  A tensor of rank 3 representing
  `activation(depthwiseconv1d(inputs, kernel) + bias)`.

Raises:
  ValueError: if `padding` is "causal".
  ValueError: when both `strides` > 1 and `dilation_rate` > 1.
c                 H   > [         TU ]  "  SUUUUUUUUU	U
UUUUUS.UD6  g )N)kernel_sizestridespaddingdepth_multiplierdata_formatdilation_rate
activationuse_biasdepthwise_initializerbias_initializerdepthwise_regularizerbias_regularizeractivity_regularizerdepthwise_constraintbias_constraint   )super__init__)selfr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   kwargs	__class__s                    l/home/james-whalen/.local/lib/python3.13/site-packages/tf_keras/src/layers/convolutional/depthwise_conv1d.pyr   DepthwiseConv1D.__init__y   sQ    & 		
#-#'!"7-"7-!5!5+!	
" #	
    c                    U R                   S:X  a  SU R                  S-  -   S-   nSnOSU R                  S-  -   nSn[        R                  " X5      n[        R                  " U R                  SS9nSU R
                  -   n[        R                  R                  UUUU R                  R                  5       U[        R                  " U R                   SS	9S
9nU R                  (       aF  [        R                  R                  UU R                  [        R                  " U R                   SS	9S9n[        R                  " Xc/5      nU R                   b  U R!                  U5      $ U$ )Nchannels_lastr      r   )r   r   r   )axis   )ndim)r   r   	dilationsr   )r   )r   r   tfexpand_dimsdepthwise_kernelr   nndepthwise_conv2dr   upperr   convert_data_formatr   bias_addbiassqueezer   )r   inputsr   spatial_start_dimr-   r   outputss          r!   callDepthwiseConv1D.call   s>   .T\\A--4G !t||a//G !:>>$*?*?aHt111%%((LL&&(#"66  q ) 	
 ==eenn		&::$$1 % G **W&9:??&??7++r#   c                    U R                   S:X  a  US   nUS   U R                  -  nO'U R                   S:X  a  US   nUS   U R                  -  n[        R                  " WU R                  S   U R
                  U R                  S   U R                  S   5      nU R                   S:X  a  US   WU4$ U R                   S:X  a  US   UW4$ g )Nchannels_firstr&   r   r%   r   )r   r   r   conv_output_lengthr   r   r   r   )r   input_shape	input_dimout_filterss       r!   compute_output_shape$DepthwiseConv1D.compute_output_shape   s    //#AI%a.4+@+@@K0#AI%a.4+@+@@K11QLLLLOq!
	 //NK;;0NI{;; 1r#    )r   validr   Nr   NTglorot_uniformzerosNNNNN)__name__
__module____qualname____firstlineno____doc__r   r8   r   shape_type_conversionr@   __static_attributes____classcell__)r    s   @r!   r   r      s\    Y| . "!!!%
N$L ##< $<r#   r   )rJ   tensorflow.compat.v2compatv2r+   5tf_keras.src.layers.convolutional.base_depthwise_convr   tf_keras.src.utilsr   r    tensorflow.python.util.tf_exportr   r   rB   r#   r!   <module>rT      sD    & " ! O ) ' : ,-|<m |< .|<r#   