[英]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.