From 138a215618d5d92ed00f7bfd4161ff5c3cdd11e4 Mon Sep 17 00:00:00 2001 From: Dosier Date: Thu, 29 Sep 2022 08:16:57 +0200 Subject: [PATCH] Added shape docs for AvgPool2D layer from Keras (#211) https://keras.io/api/layers/pooling_layers/average_pooling2d/ --- .../jetbrains/kotlinx/dl/api/core/layer/pooling/AvgPool2D.kt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/AvgPool2D.kt b/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/AvgPool2D.kt index bf24c9028..05e2559d1 100644 --- a/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/AvgPool2D.kt +++ b/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/AvgPool2D.kt @@ -14,7 +14,9 @@ import org.tensorflow.op.Ops /** * Average pooling layer for 2D inputs (e.g. images). * - * NOTE: Works with tensors which must have rank 4 (batch, height, width, channels). + * __Input shape:__ 4D tensor with shape `(batch_size, rows, cols, channels)`. + * + * __Output shape:__ 4D tensor with shape `(batch_size, pooled_rows, pooled_cols, channels)`. * * @property [poolSize] The size of the sliding window for each dimension of input tensor (pool batch, pool height, pool width, pool channels). * Usually, pool batch and pool channels are equal to 1.