Fix the doc error of step operator (#7120)

xhcao · web-flow · commit 6dccb6a7cbc4 · 2022-12-01T16:54:54.000+08:00
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'}
  */

Original file line number	Diff line number	Diff line change
`@@ -26,15 +26,15 @@ import {TensorLike} from '../types';`
`26`	`26`	`import {op} from './operation';`
`27`	`27`
`28`	`28`	`/**`
`29`		- * Computes step of the input `tf.Tensor` element-wise: `x > 0 ? 1 : alpha * x`
	`29`	+ * Computes step of the input `tf.Tensor` element-wise: `x > 0 ? 1 : alpha`
`30`	`30`	`*`
`31`	`31`	* ```js
`32`	`32`	`* const x = tf.tensor1d([0, 2, -1, -3]);`
`33`	`33`	`*`
`34`	`34`	`* x.step(.5).print(); // or tf.step(x, .5)`
`35`	`35`	* ```
`36`	`36`	`* @param x The input tensor.`
`37`		`- * @param alpha The gradient when input is negative.`
	`37`	`+ * @param alpha The gradient when input is negative. Defaults to 0.`
`38`	`38`	`*`
`39`	`39`	`* @doc {heading: 'Operations', subheading: 'Basic math'}`
`40`	`40`	`*/`