Tensorflow.js tf.prelu()函数

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

.prelu()函数用于查找所述张量输入的泄漏校正线性以及参数 alpha，并按元素完成。

句法 ：

tf.prelu(x, alpha)

哪里，x < 0 ？阿尔法 * x : f(x) = x

参数：

• x：表示张量输入，可以是 tf.Tensor、TypedArray 或 Array 类型。
• alpha：它是张量输入中规定的负值的规定比例因子，它可以是 tf.Tensor、TypedArray 或 Array 类型。

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

示例 1：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input elements
const y = tf.tensor1d([-11, 12, -31, 36]);
  
// Defining alpha
const alp = tf.scalar(0.2);
  
// Calling prelu() method and
// Printing output
y.prelu(alp).print();

输出：

Tensor
    [-2.2, 12, -6.2000003, 36]

示例 2：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling prelu() method and
// Printing output
var res = tf.prelu(tf.tensor1d(
  [-1.9, 8.2, -6.1, 13.6]), tf.scalar(-1));
res.print();

输出：

Tensor
    [1.9, 8.1999998, 6.0999999, 13.6000004]

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