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

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

用法:

tf.ones(shape, dtype, name)

参数：

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

返回值：它返回具有相同数据类型的张量。

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

// 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：在此示例中，我们将创建一个一维张量，其所有值都设置为float数据类型的1。

// 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