Skip to content

Commit

Permalink
Added shape docs for SeparableConv2D 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 7521fdc commit b1e48be
Showing 1 changed file with 5 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,11 @@ import kotlin.math.roundToInt
* Intuitively, separable convolutions can be understood as
* a way to factorize a convolution kernel into two smaller kernels, or as an extreme version of an Inception block.
*
* __Input shape:__ 4D tensor with shape `(batch_size, rows, cols, channels)`.
*
* __Output shape:__ 4D tensor with shape `(batch_size, new_rows, new_cols, filters)`.
* `rows` and `cols` values might have changed due to padding.
*
* @property [filters] The dimensionality of the output space (i.e. the number of filters in the convolution).
* @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.
Expand Down

0 comments on commit b1e48be

Please sign in to comment.