如何在pytorch中將3D張量轉換為2D張量？
[英]How to convert 3D tensor to 2D tensor in pytorch?
問題描述
我是pytorch的新手。 我有3D張量（32,10,64），我想要2D張量（32，64）。 我嘗試了view()並在將其轉換為（32,10）的線性層squeeze()后使用。
1 個解決方案
解決方案1
2 已采納 2018-11-26 05:14:25
嘗試這個
t = torch.rand(32, 10, 64).permute(0, 2, 1)[:, :, -1]
或者如Shai所指出的，您也可以
t = torch.rand(32, 10, 64)[:, -1, :]
print(t.size()) # torch.Size([32, 64])
3D張量的Pytorch內積與1D張量生成2D張量
[英]Pytorch Inner Product of 3D tensor with 1D Tensor to generate 2D Tensor
列 2D 張量乘以行 2D 張量等於 3d pytorch 張量
[英]Columns 2D tenson times rows 2D tensor equals a 3d pytorch tensor
如何在不混合相應條目的情況下將2D矩陣轉換為3D張量？
[英]How to convert 2D matrix to 3D tensor without blending corresponding entries?
聲明:本站的技術帖子網頁，遵循CC BY-SA 4.0協議，如果您需要轉載，請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.