Tensorflow.js tf.zerosLike() 函数

tf.zerosLike() 函数是 Tensorflow.js 中的一个张量操作函数，用于创建一个与输入张量形状相同的新张量，其中所有元素的值为 0。

函数语法

tf.zerosLike(x)

其中 x 是一个张量。

函数示例

const input = tf.tensor([
  [1, 2, 3],
  [4, 5, 6]
]);
const zeros = tf.zerosLike(input);
zeros.print();

运行结果：

<tf.Tensor: shape=(2, 3), dtype=float32, 
  tensor=
  [[0., 0., 0.],
   [0., 0., 0.]]>

函数说明

• tf.zerosLike() 函数创建一个与输入张量形状相同、所有元素值为 0 的新张量。
• 输入张量的形状必须是一个数组，例如 [2, 3] 表示 2 行 3 列的矩阵。
• 如果输入张量的 dtype 不是浮点型，则会转换为浮点型。
• tf.zerosLike() 函数会返回一个新的张量，不会修改输入张量。
• tf.zerosLike() 函数是一个异步函数，将返回一个 Promise 对象，可以使用 await 等待结果。

函数示例

const input = tf.tensor([
  [1, 2, 3],
  [4, 5, 6]
]);
const zeros = tf.zerosLike(input);
zeros.print();

运行结果：

<tf.Tensor: shape=(2, 3), dtype=float32, 
  tensor=
  [[0., 0., 0.],
   [0., 0., 0.]]>

总结

tf.zerosLike() 函数是 Tensorflow.js 中一个方便创建新张量的函数，它可以帮助我们快速创建一个与输入张量形状相同、所有元素值为 0 的新张量。在模型的训练和预测过程中，我们经常需要创建新张量来存储各种状态和参数。因此，熟练掌握这个函数对于开发 Tensorflow.js 应用程序非常重要。