Tensorflow.js tf.cumsum()函数

TensorFlow.js 是一个用于 JavaScript 机器学习的库。它帮助开发人员使用 JavaScript 开发 ML 模型，并直接在浏览器或 Node.js 中使用 ML。

tf.cumsum()函数用于计算 tf.Tensor 沿指定轴的累积和。一个张量的排他累积和的含义是每个张量条目不包含自己的值，而只包含排他累积和中沿指定轴的前面的值。

句法：

tf.cumsum(x, axis, exclusive, reverse)

参数：此方法有四个参数，如前所述，如下所述：

• x：必须求和的输入张量。它的类型为 tf.Tensor、TypedArray 或 Array。
• 轴：我们必须沿其求和的轴。它是一个可选参数。默认值为 0。
• 排他性：它决定是否找到一个排他的累积和。它是一个布尔值，默认为 false。它是一个可选参数。
• reverse：决定是否向相反方向求和。它是一个布尔值，默认为 false。它也是一个可选参数。

返回值：返回一个tf。Tensor表示给定张量的累积和。

下面的例子演示了 tf.cumsum() 方法。

示例 1：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Creating and initializing a new variable
const x = tf.tensor([1, 2, 3, 4]);
 
// Finding the cumulative sum
const a=x.cumsum();
 
// Printing the tensor
a.print();

输出：

Tensor
   [1, 3, 6, 10]

示例 2：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Creating and initializing a new variable
const x = tf.tensor([1, 2, 3, 4]);
 
// Finding the cumulative sum
const a=x.cumsum(0,true,true);
 
// Printing the tensor
a.print();

输出：

Tensor
   [9, 7, 4, 0]

示例 3：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Creating and initializing a new variable
const x = tf.tensor([[1, 2],[3, 4]]);
 
// Finding the cumulative sum
const a=x.cumsum();
 
// Printing the tensor
a.print();

输出：

Tensor
   [[1, 2],
    [4, 6]]

示例 4：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Creating and initializing a new variable
const x = tf.tensor([[1, 2],[3, 4]]);
 
// Finding the cumulative sum
const a=x.cumsum(1);
 
// Printing the tensor
a.print();

输出：

Tensor
   [[1, 3],
    [3, 7]]

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