如何在pytorch中将3D张量转换为2D张量？

Question

我是pytorch的新手。 我有3D张量（32,10,64），我想要2D张量（32，64）。 我尝试了view()并在将其转换为（32,10）的线性层squeeze()后使用。

Answer 1

尝试这个

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