Tensorflow.js tf.onesLike()用法及代码示例

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

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

相关用法

注：本文由纯净天空筛选整理自CoderSaty大神的英文原创作品 Tensorflow.js tf.onesLike() Function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。