numpy.pad

numpy.pad(array, pad_width, mode='constant', **kwargs)

参数

array

需要进行填充的矩阵

pad_width

填充到每个轴边缘的值。\(((before_1, after_1), ……， (before_N, after_N))\)。\(N\)的值是根据前面提供的\(array\)确定的,
比如 \((before_1, after_1)\) 是在第一个轴前填充\(before_1\)列， 在第一个轴后填充\(after_1\)列。

mode

\(str\)或者函数，是一个可选参数下列字符串值或用户提供的函数之一

constant
默认参数,使用一个常量值填充

edge
使用array的边缘值填充

maximum
使用沿每个轴的全部或部分矢量的最大值填充

mean
用矢量的全部或沿每个轴的部分平均值填充。

minimum
沿着每个轴的全部或部分矢量的最小值填充。

代码示例

import numpy as np

A = np.random.randn(2, 2, 2, 2)
print(A)
print("\n\n")

B = np.pad(A, ((0, 0), (0, 0), (1, 1), (1, 1)))
print(B)

运行结果

\(A：\)

\(B：\)