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。
axis:我们必须求和的轴。它是一个可选参数。默认值为0。
exclusive:它决定是否找到一个排他的累加总和。它是一个布尔值，默认为false。它是一个可选参数。
reverse:它决定是否求和的方向相反。它是一个布尔值，默认为false。它也是一个可选参数。

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

以下示例演示了tf.cumsum()方法。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating and intializing 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 intializing 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 intializing 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 intializing 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

相关用法

注：本文由纯净天空筛选整理自CoderSaty大神的英文原创作品 Tensorflow.js tf.cumsum() Function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。