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

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

tf.pad() 函数用于使用给定值和填充填充 tf.Tensor。

用法:

tf.pad(tensor, paddings, constantValue)

参数：此函数接受以下三个参数：

• tensor:它是一个要填充的张量。
• paddings:它是一个长度为 R 的数组，给定张量的秩，其中每个元素的长度为 2 个整数 ([pad_Before, pad_After])，指定应沿张量的每个维度给出多少填充。
• constantValue:它是要使用的填充值。默认值为 0。

返回值：它返回tf.Tensor对象。

范例1：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Intializing 3d tensor and then using 
// .pad() function to print the result
tf.tensor3d([1, 2, 3, 4], [2, 2, 1])
  .pad([[0, 0], [1, 1], [2, 2]])
      .print();

输出：

Tensor
    [[[0, 0, 0, 0, 0],
      [0, 0, 1, 0, 0],
      [0, 0, 2, 0, 0],
      [0, 0, 0, 0, 0]],

     [[0, 0, 0, 0, 0],
      [0, 0, 3, 0, 0],
      [0, 0, 4, 0, 0],
      [0, 0, 0, 0, 0]]]

范例2：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Intializing 2d tensor
let geek1 = tf.tensor2d([[1, 2], [3, 4]]);
  
// Using .pad() function.
let geek2 = geek1.pad([[0, 1], [2, 1]]);
  
// Printing the result.
geek2.print();

输出：

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

参考：https://js.tensorflow.org/api/3.6.0/#pad