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

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

tf.setdiff1dAsync()函数用于查找两个指定数字列表之间的差异。

假设给定了两个Tensor “a”和“b”，那么此函数将返回一个Tensor out，它表示存在于“a”中的所有值，而不代表存在于“b”中的所有值。返回的张量输出按照表示“a”的相同顺序进行排序。此函数还返回存在于Tensor中的数字的索引的Tensor。

用法:

tf.setdiff1dAsync (a, b)

参数：此函数接受两个参数，如下所示：

a:它是要保留的值的一维张量。
b:它是要排除在输出中的值的一维张量。它应该具有与“a”相同的数据类型。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing two Tensors
const a = [10, 20, 30, 40, 50, 60];
const b = [20, 40, 50];
  
// Calling the .setdiff1dAsync() function over
// the above two specified Tesnors as parameters
// and returns the Tensor out and indices for
// the same
const [out, indices] = await tf.setdiff1dAsync(a, b);
  
// Getting the Tensor of values present in 
// Tensor "a" but not in "b"
out.print();
  
// Getting the Tensor of indices for the 
// returned Tensor's values
indices.print();

输出：

Tensor
   [10, 30, 60]
Tensor
   [0, 2, 5]

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using two Tensors as the parameters for the 
// .setdiff1dAsync() function and
// returns the Tensor out and indices
const [out, indices] = await 
  tf.setdiff1dAsync([0.1, 0, 7.0, 7.1], [.1, 7]);
  
// Getting the Tensor of values present in 
// Tensor first Tensor parameter
// but not in second one
out.print();
  
// Getting the Tensor of indices for the 
// returned Tensor's values
indices.print();

输出：

Tensor
   [0, 7.0999999]
Tensor
   [1, 3]

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

相关用法

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