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

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

.linspace()函數用於查找指定間隔的規則間隔的數字序列。

用法：

tf.linspace(start, stop, num)

Parameters:

start:它是要生成的係列的規定起始值，並且是類型編號。
stop:它是要生成的係列的規定的最終值，並且是類型編號。
num:它是規定的要生成的值的數量，並且是類型編號。

返回值：它返回tf.Tensor1D對象。

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining all the parameters
var strt = 3;
var stp = 20;
var num = 5;
  
// Calling linspace() method and
// Printing output
tf.linspace(strt, stp, num).print();

輸出：

Tensor
    [3, 7.25, 11.5, 15.75, 20]

範例2：對所有指定的參數使用浮點值。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling linspace() method and
// Printing output
tf.linspace(12.4, 29.7, 5.4).print();

輸出：

Tensor
    [12.3999996, 16.3318176, 20.2636356, 24.1954536, 28.1272717]

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

相關用法

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