Tensorflow.js tf.booleanMaskAsync()函数

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

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

句法 ：

tf.booleanMaskAsync(tensor, mask, axis?)

参数：

• 张量：它是指定的 ND 张量，它可以是 tf.Tensor、TypedArray 或 Array 类型。
• 掩码：它是规定的 KD 布尔张量。其中，K <= N 加上 K 应该被非主动识别。它可以是 tf.Tensor、TypedArray 或 Array 类型。
• 轴：它 是 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