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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。