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

https://keras.io/api/layers/activation_layers/elu/
dosier · Sep 29, 2022 · e41ac23 · e41ac23
1 parent a175f9b
commit e41ac23
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/activation/ELU.kt b/tensorflow/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/activation/ELU.kt
@@ -21,6 +21,11 @@ import org.tensorflow.op.Ops
  * the activation closer to zero which enable faster learning as they
  * bring the gradient to the natural gradient.
  *
+ * __Input shape__:  Arbitrary. Use the keyword argument `input_shape`
+ *                   (tuple of integers, does not include the samples axis)
+ *                   when using this layer as the first layer in a model.
+ * __Output shape__: Same shape as the input.
+ *
  * @property [alpha] Hyperparameter that controls the value to which
  * an ELU saturates for negative net inputs. Should be > 0.
  * @constructor Creates [ELU] object.