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

Tensorflow.js是一個非常著名的機器學習庫，用於使用JavaScript開發機器學習模型。使用該庫的主要目的是直接從瀏覽器或在Node.js中運行和部署機器學習模型。 Tensorflow.js是一個開源的hardware-accelerated JavaScript庫。在本文中，我們將討論Tensorflow.js中的tf.ones()函數。

tf.ones()創建一個將所有元​​素設置為1的張量，或者使用值1初始化張量。

用法:

tf.ones (shape, dtype)

參數：

• Shape：這代表結果數組的形狀。形狀是一個整數數組，代表許多行和列。
• dtype：這些是返回結果的值的類型。類型的默認值為float。它可以是int32，complex64，bool，string或float32。

返回類型：

此方法返回形狀為order(row * column)且類型為1的dtype類型的張量。

範例1：在這個例子中，我們將創建一個階數為3 * 4的張量，並在其上應用tf.ones()。

//import tensorflow.js
const tf=require("@tensorflow/tfjs")
//use tf.ones()
var GFG=tf.ones([3, 4]);
  
//print tensor
GFG.print()

輸出：

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

範例2：在此示例中，我們將通過在形狀數組中提供單個元素來創建數量級為1×4的張量。

//import tensorflow.js
const tf=require("@tensorflow/tfjs")
  
//create tensor of shape 1*4
var GFG=tf.ones([3]);
  
//print tensor
GFG.print()

輸出：

Tensor
   [1, 1, 1]

範例3：tf.ones()具有不同的dtype。

//import tensorflow.js
const tf=require("@tensorflow/tfjs")
  
// create tensor with default dtype as float32
tf.ones([3, 3]).print();
  
// create tensor with complex values
tf.ones([3, 3],'complex64').print();
  
// create tensor with boolean values by default all
// values true because intialization by ones
tf.ones([3, 3],'bool').print();
  
//create tensor with integer values
tf.ones([3, 3],'int32').print();

輸出：

Tensor
   [[1, 1, 1],
    [1, 1, 1],
    [1, 1, 1]]
Tensor
   [[1 + 0j, 1 + 0j, 1 + 0j],
    [1 + 0j, 1 + 0j, 1 + 0j],
    [1 + 0j, 1 + 0j, 1 + 0j]]
Tensor
   [[true, true, true],
    [true, true, true],
    [true, true, true]]
Tensor
   [[1, 1, 1],
    [1, 1, 1],
    [1, 1, 1]]

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