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