TensorFlow.js tf.tensor() 函数

tf.tensor() 是 TensorFlow.js 中用于创建张量（Tensor）的函数之一。张量是 TensorFlow.js 中的核心数据结构，类似于多维数组，用于存储和执行数值计算。

语法

tf.tensor(values, shape?, dtype?)

• values：表示张量的值，可以是一个一维或多维数组，也可以是一个TypedArray。
• shape（可选）：表示张量的形状，可以是一个数字数组。默认情况下，形状将根据 values 参数自动推导。
• dtype（可选）：表示张量的数据类型，可以是 'float32'、'int32'、'bool' 等。默认情况下，数据类型将根据 values 参数自动推导。

返回值

• 返回一个张量对象。

示例

创建一个一维张量

const tensor1D = tf.tensor([1, 2, 3]);

创建一个二维张量

const tensor2D = tf.tensor([[1, 2], [3, 4]]);

创建一个带有指定形状和数据类型的张量

const shape = [2, 2];  // 形状为 2x2
const dtype = 'int32'; // 数据类型为 int32
const tensor = tf.tensor([1, 2, 3, 4], shape, dtype);

说明

• 使用 tf.tensor() 函数可以从 JavaScript 数组或 TypedArray 创建张量。
• 当 shape 和 dtype 参数未指定时，tf.tensor() 函数将根据 values 参数自动推导出张量的形状和数据类型。
• 张量对象是不可变的，它们的值不能被修改。如果需要进行计算操作，可以使用 TensorFlow.js 提供的其他函数。

支持的数据类型

• 'float32'（默认）
• 'int32'
• 'bool'

注意：TensorFlow.js tf.tensor() 函数还接受其他参数，用于更详细地控制张量的创建过程。详情请参阅 TensorFlow.js 文档。

参考资料

• TensorFlow.js 官方文档：tf.tensor()
• TensorFlow.js 官方 GitHub 仓库：tensorflow/tfjs