如何转换PyTorch张量维度?

问题描述 投票:0回答:1

我使用 PyTorch 得到尺寸为 (210,2) 的张量。我想把它转换成(15,14,2)。 有人知道怎么做吗?先谢谢你了

pytorch tensor
1个回答
0
投票
x = torch.randn(210, 2)
x = x.reshape(15,14,2)
© www.soinside.com 2019 - 2024. All rights reserved.