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

Tensorflow.js是由Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型以及深度學習神經網絡。

.dilation2d() 函數用於評估指定輸入張量的灰度膨脹。

用法:

tf.dilation2d(x, filter, strides, pad, dilations?, dataFormat?)

參數：

• x:指定的輸入張量是 3 級或 4 級，形狀為：[batch, height, width, inChannels]。此外，如果等級為 3，則假定批次大小為 1。它可以是 tf.Tensor3D、tf.Tensor4D、TypedArray 或 Array 類型。
• filter:所述的 3 級濾波器張量和形狀：[filterHeight, filterWidth, depth]。它可以是 tf.Tensor3D、TypedArray 或 Array 類型。
• strides:給定輸入張量的每個尺寸的滑動窗口的規定步幅：[strideHeight, strideWidth]。如果規定的步幅是單個數字，則 strideHeight == strideWidth。它可以是 [number, number] 或 number 類型。
• pad:規定的填充算法類型。它可以是有效或相同的類型。
  1. 在這裏，對於相同和步長 1，無論過濾器大小如何，輸出都將具有與輸入相同的大小。
  2. 對於，‘valid’，在濾波器尺寸大於1*1×1的情況下，輸出應小於輸入。
• dilations:規定的擴張率：[dilationHeight, dilationWidth] 因為輸入值在高度和寬度維度上采樣，有利於多孔形態擴張。默認值為 [1, 1]。此外，如果 dilation 是單個數字，則 dilationHeight == dilationWidth。如果它大於 1，則步長的所有值都應為 1。它是可選的，類型為 [number, number], number。
• dataFormat:它指定了所述輸入和輸出數據的數據格式。默認值為“NHWC”。而且，這裏的數據是按照以下順序存儲的：[batch, height, width, channels]。它是可選的，屬於“NHWC”類型。

返回值：它返回 tf.Tensor3D 或 tf.Tensor4D。

範例1：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining input tensor
const x = tf.tensor3d([1, 2, 3, 4], [2, 2, 1]);
  
// Defining filter tensor
const y = tf.tensor3d([1, 1, 0, 4], [1, 1, 4]);
  
// Calling dilation2d() method
const result = tf.dilation2d(x, y, 2, 'valid');
  
// Printing output
result.print();

輸出：

Tensor
     [ [[2],]]

範例2：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling dilation2d() method with 
// all its parameters
tf.tensor3d([1.1, 2.2, 3.3, 4.4], [2, 2, 1]).dilation2d(
 tf.tensor3d([1.3, 1.2, null, -4], [1, 1, 4]),
             2, 'valid', [3, 2], 'NHWC').print();

輸出：

Tensor
     [ [[2.4000001],]]

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