Tensorflow.js tf.depthToSpace()用法及代碼示例

Tensorflow.js是Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。它還可以幫助開發人員使用JavaScript語言開發ML模型，並可以直接在瀏覽器或Node.js中使用ML。

tf.depthToSpace()是tensorflow.js庫的內置函數，用於重新排列輸入張量中的數據，其中深度維度中的值在空間塊中移動到高度維度和寬度維度。它將數據從深度重新排列為空間數據塊。

用法:

tensor.depthToSpace(input, blocksize, dataformat)

參數表：

input:給定張量
blocksize:輸出張量的寬度為depth * blockSize。
dataformat：指定給定和結果張量的布局。它有兩個選項：“NHWC”：[批處理，高度，寬度，通道]和“NCHW”：[批處理，通道，高度，寬度]

返回值：它返回相同數據類型的重排張量。

範例1：使用“NHWC”格式

Javascript


// Importing the tensorflow.Js lbrary
import * as tf from "@tensorflow/tfjs"
  
// Create a new tensor
var input = tf.tensor4d([1, 3, 5, 7], [1, 1, 1, 4]);
  
// define block size
var blockSize = 2;
  
// define data format
var dataFormat = "NHWC";
  
// rearrange data
var val = tf.depthToSpace(input, blockSize, dataFormat);
  
// print the tensor
val.print();

輸出

Tensor
    [[[[1],
       [3]],

      [[5],
       [7]]]]

範例2：使用“NCHW”格式

Javascript


// Importing the tensorflow.Js lbrary
import * as tf from "@tensorflow/tfjs"
  
// Create a new tensor
var input = tf.tensor4d([1, 3, 5, 7], [1, 4, 1, 1]);
  
// define block size
var blockSize = 2;
  
// define data format
var dataFormat = "NCHW";
  
// rearrange data
var tr = tf.depthToSpace(input, blockSize, dataFormat);
  
// print the tensor
tr.print();

輸出

Tensor
    [[[[1, 3],
       [5, 7]]]]

參考：https://js.tensorflow.org/api/latest/#depthToSpace

相關用法

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