首頁(yè) 猿問如何填充不同形狀的張量以使它們?cè)诙?..

如何填充不同形狀的張量以使它們?cè)诙S中具有相同的形狀

Python

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

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

查看完整描述

1 回答

眼眸繁星

TA貢獻(xiàn)1873條經(jīng)驗(yàn) 獲得超9個(gè)贊

如果您使用的是 numpy 數(shù)組，則可以通過這種方式對(duì)它們進(jìn)行零填充：

# 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])

反對(duì) 回復(fù) 2023-05-23

關(guān)注

舉報(bào)

0/150

提交

取消

購(gòu)課補(bǔ)貼
聯(lián)系客服咨詢優(yōu)惠詳情

慕課網(wǎng)APP
您的移動(dòng)學(xué)習(xí)伙伴

掃描二維碼
關(guān)注慕課網(wǎng)微信公眾號(hào)