From 6f050277f3dad1b63f7b793646c4ca2217d9264b Mon Sep 17 00:00:00 2001 From: Xinghua Cao Date: Tue, 29 Nov 2022 13:38:37 +0800 Subject: [PATCH] Fix the doc error of step operator --- tfjs-core/src/ops/step.ts | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tfjs-core/src/ops/step.ts b/tfjs-core/src/ops/step.ts index b279a2b899..26c9100262 100644 --- a/tfjs-core/src/ops/step.ts +++ b/tfjs-core/src/ops/step.ts @@ -26,7 +26,7 @@ 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]); @@ -34,7 +34,7 @@ import {op} from './operation'; * 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'} */