Tensorflow.js tf.cumsum()函数(1)

📌 相关文章

📜 Tensorflow.js tf.cumsum()函数(1)

📅 最后修改于: 2023-12-03 14:47:54.727000 🧑 作者: Mango

Tensorflow.js tf.cumsum()函数介绍

简介

Tensorflow.js是Tensorflow的JavaScript版本，用于在浏览器和Node.js中开发深度学习模型。tf.cumsum()函数是其中一个用于计算的函数，用于计算给定轴上元素的累加和。该函数返回一个张量从第一个元素开始依次计算累加和。

语法

tf.cumsum(
    x: Tensor,
    axis?: number,
    exclusive?: boolean,
    reverse?: boolean
): Tensor

参数说明：

x：输入的张量。
axis：计算累加和的轴。默认为0。
exclusive：如果是true，则在计算时不包括当前元素。默认为false。
reverse：如果是true，则从最后一个元素开始计算累加和。默认为false。

示例

Node.js

const tf = require('@tensorflow/tfjs');

const x = tf.tensor2d([[1, 2], [3, 4]]);
const res = tf.cumsum(x, 1);

res.print();

输出：

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

浏览器

<!DOCTYPE html>
<html>
  <head>
    <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@3.10.0/dist/tf.js"></script>
  </head>
  <body>
    <script>
      const x = tf.tensor2d([[1, 2], [3, 4]]);
      const res = tf.cumsum(x, 1);

      res.print();
    </script>
  </body>
</html>

输出：

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

总结

tf.cumsum()函数是Tensorflow.js中用于计算累加和的函数。它的使用非常简单，只需要传入一个张量和需要计算累加和的轴即可。在实际开发中，该函数可以用于任意需要累加和的场景。