如何填充不同形状的张量以使其在第二维上具有相同的形状

问题描述

我有3个张量分别为（100,43,1024），（100,37,1024）和（100,42,1024）。我想将所有这些张量的第二维设为第二维的最大值，在这种情况下为43。您能帮助我如何使用垫功能使它们具有相同的形状吗？

解决方法

如果您使用numpy数组，则可以通过以下方式对它们进行零填充：

# create your data
n_sample = 5
X = [np.random.uniform(0,1,(n_sample,43,1024)),np.random.uniform(0,37,42,1024))]

# find max dim
max_dim = np.max([x.shape[1] for x in X])
print(max_dim)

X_pad = []
for x in X:
    X_pad.append(np.pad(x,((0,0),(max_dim-x.shape[1],(0,0)),mode='constant')) # pre padding
    # X_pad.append(np.pad(x,max_dim-x.shape[1],mode='constant')) # post padding
    
# check padded shape
print([x.shape for x in X_pad])