Tensorflow.js tf.booleanMaskAsync()用法及代码示例

Tensorflow.js是由Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

.booleanMaskAsync() 函数用于实现指定输入张量的布尔掩码。

用法：

tf.booleanMaskAsync(tensor, mask, axis?)

Parameters:

tensor:它是规定的 N-D 张量，它可以是 tf.Tensor、TypedArray 或 Array 类型。
mask:它是规定的 K-D 布尔张量。其中，K <= N 加 K 应该是非主动识别的。它可以是 tf.Tensor、TypedArray 或 Array 类型。
axis:它是类型为 number 的可选参数。它是一个 0-D 整数类型的张量，表示要屏蔽的所述张量中的轴。在这里，默认值是零，它屏蔽了第一个尺寸，否则 (K + 轴 <= N)。

返回值：它返回 Promise(tf.Tensor)。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input and mask
const tn = tf.tensor2d(
    [7, 8, 9, 10, 11, 12, 5, 62], [4, 2]);
const msk = tf.tensor1d([2, 1, 2, 3], 'bool');
  
// Calling tf.booleanMaskAsync() method and
// Printing output
const res = await tf.booleanMaskAsync(tn, msk);
res.print();

输出：

Tensor
    [[7 , 8 ],
     [9 , 10],
     [11, 12],
     [5 , 62]]

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling tf.booleanMaskAsync() method and
// Printing output
const res = await tf.booleanMaskAsync(
    tf.tensor2d([34, 17, 7, 8], [2, 2]), 
    tf.tensor1d([1, 1], 'bool'), 1);
      
res.print();

输出：

Tensor
    [[34, 17],
     [7 , 8 ]]

参考： https://js.tensorflow.org/api/latest/#booleanMaskAsync

相关用法

注：本文由纯净天空筛选整理自nidhi1352singh大神的英文原创作品 Tensorflow.js tf.booleanMaskAsync() Function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。