TensorFlow.js 之 tf.sparseToDense() 函数

概述

tf.sparseToDense() 函数是 TensorFlow.js 中用来将稀疏张量转化为稠密张量的函数。这个函数的主要作用是将一个以稀疏形式存储的向量或矩阵转化为稠密形式，便于后续计算。

语法

tf.sparseToDense(
    sparseIndices: tf.Tensor2D | tf.Tensor1D,
    sparseValues: tf.Tensor,
    outputShape: number[],
    defaultValue?: number
) => tf.Tensor

参数

• sparseIndices：要转化为稠密张量的稀疏张量的索引。可以是二维张量或一维张量。
• sparseValues：在稀疏张量中对应于 sparseIndices 的值。
• outputShape：生成稠密张量的形状（shape）数组。
• defaultValue：生成的稠密张量中的默认值。默认值为 0。

返回值

稠密张量。

示例

二维张量的转化

const sparseTensor = tf.sparse.fromTensor([2, 2], [[0, 1], [1, 0]], [1, 2]);
console.log(sparseTensor.rank); // 2

const denseTensor = tf.sparseToDense(sparseTensor.indices, sparseTensor.values, [2, 2], 0);

denseTensor.print(); // Output: [[0, 1], [2, 0]]

一维张量的转化

const sparseTensor = tf.sparse.fromTensor([2], [0, 1], [10]);
console.log(sparseTensor.rank); // 1

const denseTensor = tf.sparseToDense(sparseTensor.indices, sparseTensor.values, [4], 0);

denseTensor.print(); // Output: [10, 0, 0, 0]

参考文献

• TensorFlow.js: tf.sparseToDense()