Added shape docs for ZeroPadding3D layer from Keras (Kotlin#211)

https://keras.io/api/layers/reshaping_layers/zero_padding3d/
dosier · Sep 29, 2022 · d5c8d88 · d5c8d88
1 parent e6d5118
commit d5c8d88
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...orflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/reshaping/ZeroPadding3D.kt b/...orflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/reshaping/ZeroPadding3D.kt
@@ -10,6 +10,11 @@ import org.tensorflow.Shape
 /**
  * Zero-padding layer for 3D input (e.g. video).
  * This layer can add zeros in the rows, cols and depth of a video tensor.
+ *
+ * __Input shape:__  5D tensor with shape `(batch_size, first_axis_to_pad, second_axis_to_pad, third_axis_to_pad, depth)`.
+ *
+ * __Output shape:__ 5D tensor with shape `(batch_size, first_padded_axis, second_padded_axis, third_axis_to_pad, depth)`.
+ *
  * @property [padding] 6 numbers  interpreted as `(left_dim1_pad, right_dim1_pad, left_dim2_pad, right_dim2_pad, left_dim3_pad, right_dim3_pad)`.
  *
  * @since 0.3