Tensorflow.js tf.ones()函数

Tensorflow.js 是一个开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。 tf.ones()函数用于创建一个新张量，其中所有元素都设置为 1。

句法：

tf.ones(shape, dtype, name)

参数：

• shape：它采用张量的形状。它可以是多维数组或 int32
• dtype：它采用我们插入的 1 的数据类型。它默认设置为 float32，但也可以是 int32。它是一个可选参数。
• name：它采用我们正在执行的操作的名称。默认为无。它是一个可选参数。

返回值：返回一个相同数据类型的张量。

示例 1：在此示例中，我们将创建一个 2D 张量，其所有值都设置为整数数据类型的 1。

JavaScript

// Import tensorflow
import tensorflow as tf
  
// Get a Tensor
val = tf.ones([10, 10], tf.int32)
  
// Print the Tensor
print(val)

输出：

tf.Tensor(
[[1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]
 [1 1 1 1 1 1 1 1 1 1]], shape=(10, 10), dtype=int32)

示例 2：在此示例中，我们将创建一个一维张量，其所有值都设置为浮点数据类型的 1。

JavaScript

// Import tensorflow
import tensorflow as tf
  
// Get a Tensor
val = tf.ones(5, tf.float32)
  
// Print a Tensor
print(val)

输出：

tf.Tensor([1. 1. 1. 1. 1.], shape=(5,), dtype=float32)

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