Skip to content

Commit

Permalink
Added shape docs for DepthwiseConv2D layer from Keras (Kotlin#211)
Browse files Browse the repository at this point in the history
  • Loading branch information
dosier committed Sep 29, 2022
1 parent aa8f4f3 commit 7521fdc
Showing 1 changed file with 5 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,11 @@ import org.tensorflow.op.nn.DepthwiseConv2dNative
* The `depthMultiplier` argument controls how many
* output channels are generated per input channel in the depthwise step.
*
* __Input shape:__ 4D tensor with shape `(batch_size, rows, cols, channels)`.
*
* __Output shape:__ 4D tensor with shape `(batch_size, new_rows, new_cols, channels * depth_multiplier)`.
* `rows` and `cols` values might have changed due to padding.
*
* @property [kernelSize] Two long numbers, specifying the height and width of the 2D convolution window.
* @property [strides] Strides of the pooling operation for each dimension of input tensor.
* NOTE: Specifying any stride value != 1 is incompatible with specifying any `dilation_rate` value != 1.
Expand Down

0 comments on commit 7521fdc

Please sign in to comment.