将ndarray 有维度(1,222,1024) 扩充到(1,256,1024)使用numpy.pad()函数

将ndarray 有维度(1,222,1024) 扩充到(1,256,1024)使用numpy.pad()函数
可以使用numpy.pad()函数将ndarray从维度(1, 222, 1024)扩充到(1, 256, 1024)。numpy.pad()函数的语法为:

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

其中,array是要扩充的数组,pad_width是一个元组,用于指定每个轴的填充宽度,mode是填充模式。

以下是将ndarray从维度(1, 222, 1024)扩充到(1, 256, 1024)的代码:

Copy code
import numpy as np

# 创建一个维度为(1,222,1024)的随机数组
arr = np.random.rand(1, 222, 1024)

# 定义填充宽度
pad_width = ((0, 0), (17, 17), (0, 0))

# 使用numpy.pad()函数进行填充
padded_arr = np.pad(arr, pad_width, mode='constant')

# 打印填充后数组的维度
print(padded_arr.shape)

运行代码后,将输出填充后数组的维度为(1, 256, 1024)。其中,pad_width的第一个元素(0, 0)表示在第一个轴不进行填充,第二个元素(17, 17)表示在第二个轴两侧各填充17个元素,第三个元素(0, 0)表示在第三个轴不进行填充。

猜你喜欢

转载自blog.csdn.net/qq_41563601/article/details/129420718