revised layer_normalization.js

webmachinelearning · Dec 8, 2023 · e6b9d71 · e6b9d71
1 parent abd89a9
commit e6b9d71
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/layer_normalization.js b/src/layer_normalization.js
@@ -24,8 +24,7 @@ export function getIndexOfSortedValue(axes) {
  * Normalize the tensor values of input features using
  * [layer-Normalization](https://arxiv.org/abs/1607.06450)
  * @param {Tensor} input
- * @param {Array} axes
- * @param {MLBatchNormalizationOptions} [options]
+ * @param {MLLayerNormalizationOptions} [options]
  * @return {Tensor}
  */
 export function layerNormalization(input, {scale, bias, axes, epsilon=1e-5}) {