如何填充不同形状的张量以使它们在二维中具有相同的形状

Question

我有 3 个张量，形状为 (100,43,1024)、(100,37,1024) 和 (100,42,1024)。我想使所有这些张量的第二维成为第二维的最大值，即在这种情况下为 43。你能帮我看看如何使用 pad 功能使它们形状相同吗？

Answer 1

如果您使用的是 numpy 数组，则可以通过这种方式对它们进行零填充：# create your datan_sample = 5X = [np.random.uniform(0,1, (n_sample,43,1024)),     np.random.uniform(0,1, (n_sample,37,1024)),     np.random.uniform(0,1, (n_sample,42,1024))]# find max dimmax_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,0)), mode='constant')) # pre padding    # X_pad.append(np.pad(x, ((0,0),(0,max_dim-x.shape[1],(0,0)), mode='constant')) # post padding    # check padded shapeprint([x.shape for x in X_pad])