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

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

tf.whereAsync()函數用於返回指定條件下真實元素的二維張量。在返回的張量中，第一維(即行)指定了真實元素的數量，第二維(即列)指定了真實元素的坐標，即輸出張量的形狀為[numTrueElems，condition.rank]。

注意：返回的張量的形狀取決於輸入中存在的真實值。

用法：

tf.whereAsync (condition)

參數：此函數接受如下所示的參數：

condition:輸入條件是布爾值數組。

返回值：它返回指定條件下真實元素的二維張量。

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a tensor for conditions
const cond = tf.tensor1d([false, true, true, false, true], 'bool');
  
// Calling the .whereAsync() function
const result = await tf.whereAsync(cond);
  
// Getting the 2-D tensor of true elements
result.print();

輸出：

Tensor
   [[1],
    [2],
    [4]]

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using a tensor of conditions as the
// parameter of the .whereAsync() function
const result = await tf.whereAsync(tf.tensor1d(
  [true, false, true, true, false], 'bool'));
  
// Getting the 2-D tensor of true elements
result.print();

輸出：

Tensor
   [[0],
    [2],
    [3]]

參考：https://js.tensorflow.org/api/1.0.0/#whereAsync

相關用法

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