Tensorflow.js tf.relu()函数
Tensorflow.js 是由谷歌开发的开源库,用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。
.relu()函数用于查找所述张量输入即 max(x, 0) 的整流线性,并且按元素完成。
句法 :
tf.relu(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([11, 76, 0, -4, 6]);
// Calling relu() method and
// Printing output
y.relu().print();
Javascript
// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Defining tensor input
var val = [1.5, -5, .22, null, 'a'];
// Calling tensor1d method
const y = tf.tensor1d(val);
// Calling relu() method
var res = tf.relu(y)
// Printing output
res.print();
输出:
Tensor
[11, 76, 0, 0, 6]
示例 2:
Javascript
// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Defining tensor input
var val = [1.5, -5, .22, null, 'a'];
// Calling tensor1d method
const y = tf.tensor1d(val);
// Calling relu() method
var res = tf.relu(y)
// Printing output
res.print();
输出:
Tensor
[1.5, 0, 0.22, 0, NaN]
参考: https://js.tensorflow.org/api/latest/#relu