簡體   English   中英

如何將 pytorch 張量從 [a, b*c] 重塑為 [b,a,c]?

[英]How do I reshape pytorch tensor from [a, b*c] to [b,a,c]?

我收到此代碼錯誤,我不確定如何重塑我的張量。

RuntimeError: Expected 3-dimensional input for 3-dimensional weight [32, 35, 2], but got 2-dimensional input of size [35, 64] instead

您可以使用來自火炬的reshape function 如下:

a = torch.rand((35,64))
b = torch.reshape(a, (32,35,2))

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM