Updated shape documentation in GlobalAvgPool2D (Kotlin#211)

- added emphasis - removed redundant note comment
dosier · Sep 29, 2022 · a175f9b · a175f9b
1 parent 71a87e8
commit a175f9b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/...orflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/GlobalAvgPool2D.kt b/...orflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/GlobalAvgPool2D.kt
@@ -13,11 +13,9 @@ import org.tensorflow.op.Ops
 /**
  * Global average pooling operation for 2D data (images and so on).
  *
- * NOTE: Works with tensors which must have rank 4 (batch, height, width, channels).
+ * __Input shape:__  4D tensor with shape `(batch_size, rows, cols, channels)`.
  *
- * Input shape: 4D tensor with shape `(batch_size, rows, cols, channels)`.
- *
- * Output shape: 2D tensor with shape `(batch_size, channels)`.
+ * __Output shape:__ 2D tensor with shape `(batch_size, channels)`.
  *
  * @property [name] Custom layer name.
  * @constructor Creates [GlobalAvgPool2D] object.