TensorFlow.js tf.fill()函数

TensorFlow.js中的tf.fill()函数以给定的形状和标量值创建了一个张量。它可以用来创建矩阵和向量。

下面是该函数的语法：

tf.fill(shape, value, dtype?)

参数说明如下：

• shape：Tensor的形状，必须是一个数字数组；
• value：填充张量的标量值；
• dtype：（可选参数）创建张量的数据类型，默认为 float32。

该函数返回一个新的Tensor。

以下是一个填充矩阵的示例：

const input = tf.fill([3, 2], 0.5);

input.print();
// 输出矩阵：
// Tensor
//  [[0.5, 0.5],
//   [0.5, 0.5],
//   [0.5, 0.5]]

以下是一个填充向量的示例：

const input = tf.fill([5], 1);

input.print();
// 输出向量：
// Tensor [1, 1, 1, 1, 1]

在这个例子中，我们使用了fill()函数来创建一个长度为5的向量，并填充了它的每个元素。

如果您指定一个数据类型作为可选参数，会创建一个具有指定数据类型的张量。

以下是一个使用int32类型创建张量的示例：

const input = tf.fill([2, 2], 42, 'int32');

input.print();
// 输出矩阵：
// Tensor
//  [[42, 42],
//  [42, 42]], dtype: 'int32'

在这个例子中，我们使用了fill()函数来创建一个 $2 \times 2$ 的张量，并使用 int32 类型来填充它的每个元素。

tf.fill()函数是很实用的函数，可以让我们快速创建填充特定值的张量。