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

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

.scatterND()函数用于根据声明的索引，通过对声明的形状张量的零张量内的单个切片或值进行分散更新，来形成不同的张量。此外，此函数是对tf.gatherND()函数的取反，该函数从指定的张量中获取切片或值。

用法：

tf.scatterND(indices, updates, shape)

Parameters:

indices:指定的张量保存指向输出张量的索引，并且可以是tf.Tensor，TypedArray或Array类型。
updates:它是声明的张量，用于保存索引的值，并且可以是tf.Tensor，TypedArray或Array类型。
shape:它是输出张量的规定形状，类型为number []。

返回值：它返回tf.Tensor对象。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining indices, updates and shape
const ind = tf.tensor2d([6, 5, 2], [3, 1], 'int32');
const updat = tf.tensor1d([1, 2, 3]);
const shp = [6];
  
// Calling tf.scatterND() method and
// Printing output
tf.scatterND(ind, updat, shp).print();

输出：

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

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling tf.scatterND() method and
// Printing output
tf.scatterND(tf.tensor2d([5.4, 2.4], [2, 1], 'int32'), 
            tf.tensor1d([1.8, 4.2]), 
            [4]).print();

输出：

Tensor
    [0, 0, 4.1999998, 0]

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

相关用法

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