當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


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:

Javascript


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:

Javascript


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

相關用法


注:本文由純淨天空篩選整理自taran910大神的英文原創作品 Tensorflow.js tf.sparseFillEmptyRows() Function。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。