Tensorflow.js tf.relu6()函数

Tensorflow.js 是一个由谷歌开发的开源库，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

.relu6()函数用于查找校正后的线性 6 即 min(max(x, 0), 6) 并按元素完成。

句法 ：

tf.relu6(x)

参数：

• x：表示张量输入，可以是 tf.Tensor、TypedArray 或 Array 类型。此外，如果指定的数据类型是布尔类型，那么输出数据类型将是 int32 类型。

返回值：它返回 tf.Tensor 对象。

示例 1：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input elements
const y = tf.tensor1d([1, 176, 0, NaN, -4]);
  
// Calling relu6() method and
// Printing output
y.relu6().print();

输出：

Tensor
    [1, 6, 0, NaN, 0]

示例 2：

Javascript

// Importing the tensorflow.js library 
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input
var val = [9.5, .4, "abc", null, 'z'];
  
// Calling tensor1d method
const y = tf.tensor1d(val);
  
// Calling relu6() method
var res = tf.relu6(y)
  
// Printing output
res.print();

输出：

Tensor
    [6, 0.4, NaN, 0, NaN]

参考： https://js.tensorflow.org/api/latest/#relu6