已解決430363個問題，去搜搜看，總會有你想問的

如何填充不同形狀的張量以使它們在二維中具有相同的形狀

關注

首頁猿問如何填充不同形狀的張量以使它們在二...

Python

12345678_0001 2023-05-23 15:06:59

我有 3 個張量，形狀為 (100,43,1024)、(100,37,1024) 和 (100,42,1024)。我想使所有這些張量的第二維成為第二維的最大值，即在這種情況下為 43。你能幫我看看如何使用 pad 功能使它們形狀相同嗎？

查看完整描述

1 回答

眼眸繁星

TA貢獻1873條經驗獲得超9個贊

如果您使用的是 numpy 數組，則可以通過這種方式對它們進行零填充：

# create your data

n_sample = 5

X = [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 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,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 shape

print([x.shape for x in X_pad])

反對回復 2023-05-23

關注

舉報

0/150

提交

取消