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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。