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

Tensorflow.js是Google开发的一个开放源代码库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。。它还可以帮助开发人员使用JavaScript语言开发ML模型，并且可以直接在浏览器或Node.js中使用ML。

tf.sparseFillEmptyRows()函数用于获取通过输入图表示的输入SparseTensor，(索引，值，densityShape)。

用法:

tf.sparseFillEmptyRows(indices, values, denseShape, defaultValue)

参数：

• indices:稀疏张量的索引。
• values:稀疏张量的值。
• denseShape:稀疏张量的形状。
• defaultValue:要插入位置的默认值。

返回值：它返回tf.Tensor。

范例1：

const result = tf.sparse.sparseFillEmptyRows(
    [[0, 1], [1, 2], [2, 3], [3, 4], [4, 5]],
    [0, 1, 2, 3, 4], [5, 6], -1);
  
result['outputIndices'].print();
result['outputValues'].print();
result['reverseIndexMap'].print();

输出：

Tensor
    [[0, 1],
     [1, 2],
     [2, 3],
     [3, 4],
     [4, 5]]
Tensor
    [0, 1, 2, 3, 4]
Tensor
    [0, 1, 2, 3, 4]

范例2：

const result = tf.sparse.sparseFillEmptyRows(
    [[1], [1], [4], [3], [2]],
    [4, 6, 8, 5, 3], [7, 2], -1);
  
console.log(result);

输出：

{
  "outputIndices":{
    "kept":false,
    "isDisposedInternal":false,
    "shape":[
      5,
      1
    ],
    "dtype":"float32",
    "size":5,
    "strides":[
      1
    ],
    "dataId":{
      "id":12
    },
    "id":12,
    "rankType":"2",
    "scopeId":2
  },
  "outputValues":{
    "kept":false,
    "isDisposedInternal":false,
    "shape":[
      5
    ],
    "dtype":"float32",
    "size":5,
    "strides":[],
    "dataId":{
      "id":13
    },
    "id":13,
    "rankType":"1",
    "scopeId":2
  },
  "emptyRowIndicator":{
    "kept":false,
    "isDisposedInternal":false,
    "shape":[
      7
    ],
    "dtype":"bool",
    "size":7,
    "strides":[],
    "dataId":{
      "id":14
    },
    "id":14,
    "rankType":"1",
    "scopeId":2
  },
  "reverseIndexMap":{
    "kept":false,
    "isDisposedInternal":false,
    "shape":[
      5
    ],
    "dtype":"float32",
    "size":5,
    "strides":[],
    "dataId":{
      "id":15
    },
    "id":15,
    "rankType":"1",
    "scopeId":2
  }
}

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