Tensorflow.js tf.onesLike()函数

Tensorflow.js 是谷歌开发的一个开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。

tf.onesLike()函数用于创建一个tf.Tensor ，其中所有元素都设置为 1，其形状与给定的张量相同。

句法：

tf.onesLike (x)

参数：

• x：一个 tf.Tensor，其所有元素都将设置为 1。

返回值：它返回一个 tf.Tensor。

示例 1：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor
const x = tf.tensor([1, 2]);
  
// Printing the tensor after setting the value of tensor to 1
tf.onesLike(x).print();

输出：

Tensor
   [1, 1]

示例 2：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor
const x = tf.tensor2d([8, 2, 5, 6], [2, 2]);
  
// Printing the tensor after setting the value of tensor to 1
tf.onesLike(x).print();

输出：

Tensor
   [[1, 1],
    [1, 1]]

参考： https://js.tensorflow.org/api/latest/#onesLike