Tensorflow.js tf.pad()函数

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

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

句法：

tf.pad(tensor, paddings, constantValue)

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

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

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

示例 1：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Initializing 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：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Initializing 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