Fix the doc error of step operator

tensorflow · Dec 1, 2022 · 6f05027 · 6f05027
1 parent 9d28deb
commit 6f05027
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tfjs-core/src/ops/step.ts b/tfjs-core/src/ops/step.ts
@@ -26,15 +26,15 @@ import {TensorLike} from '../types';
 import {op} from './operation';
 
 /**
- * Computes step of the input `tf.Tensor` element-wise: `x > 0 ? 1 : alpha * x`
+ * Computes step of the input `tf.Tensor` element-wise: `x > 0 ? 1 : alpha`
  *
  * ```js
  * const x = tf.tensor1d([0, 2, -1, -3]);
  *
  * x.step(.5).print();  // or tf.step(x, .5)
  * ```
  * @param x The input tensor.
- * @param alpha The gradient when input is negative.
+ * @param alpha The gradient when input is negative. Defaults to 0.
  *
  * @doc {heading: 'Operations', subheading: 'Basic math'}
  */