diff --git a/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/AvgPool1D.kt b/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/AvgPool1D.kt
index 58a6afb32..8082d6036 100644
--- a/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/AvgPool1D.kt
+++ b/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/AvgPool1D.kt
@@ -17,6 +17,10 @@ import org.tensorflow.op.core.Squeeze
  *
  * Downsamples the input by taking the average over a temporal window of size [poolSize].
  *
+ * __Input shape:__  3D tensor with shape `(batch_size, steps, features)`.
+ *
+ * __Output shape:__ 3D tensor with shape `(batch_size, downsampled_steps, features)`.
+ *
  * @property [poolSize] Size of the temporal pooling window for each dimension of input.
  * @property [strides] The amount of shift for pooling window per each input dimension in each pooling step.
  * @property [padding] Padding strategy; can be either of [ConvPadding.VALID] which means no