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

Tensorflow.js是Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。

tf。 range()用於創建一個新的tf.Tensor1D，其中填充了借助start，stop，step和dtype提供的範圍內的數字。

用法:

tf.range(start, stop, step, dtype)

參數：

start:它是一個整數起始值，表示範圍的起始編號。
stop:它是一個整數終止值，表示範圍的結束號，並且不包括在內。
step:它是一個整數增量，默認情況下為1或-1。它是一個可選參數。
dtype:它是輸出張量的數據類型。默認為‘float32’。它是一個可選參數。

返回值：它返回一個新的Tensor1D對象。

範例1：在此示例中，我們嘗試使用默認步驟1生成一個範圍為1到9的數字。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor1D array by tf.range
var val = tf.range(1,10);
  
// Printing the tensor
val.print()

輸出：

Tensor
    [1, 2, 3, 4, 5, 6, 7, 8, 9]

範例2：在此示例中，我們嘗試使用步長2生成1到10之間的奇數。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor1D array by tf.range
var val = tf.range(1,10,2);
  
// Printing the tensor
val.print()

輸出：

Tensor
    [1, 3, 5, 7, 9]

範例3：在此示例中，我們嘗試使用步長2生成0到10之間的偶數。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor1D array by tf.range
var val = tf.range(0,10,2);
  
// Printing the tensor
val.print()

輸出：

Tensor
    [0, 2, 4, 6, 8],

範例4：在此示例中，我們將嘗試使用dtype參數。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor1D array by tf.range
var val = tf.range(-1,1,1,'bool');
  
// Printing the tensor
val.print()

輸出：

Tensor
    [true, false]

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

相關用法

注：本文由純淨天空篩選整理自121910316053大神的英文原創作品 Tensorflow.js tf.range() Function。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。