基于 pytorch,练习 tensor 常见操作 修改 x 的维度为(3,4);
时间: 2023-09-19 15:03:35 浏览: 87
PyTorch中Tensor的维度变换实现
使用pytorch库中的tensor类,可以很方便地进行维度的修改操作。对于给定的一个tensor x,我们可以使用x.view()方法来修改它的维度。假设x的大小为(2, 6),我们想要将其维度修改为(3, 4),可以按照以下步骤进行操作:
1. 导入pytorch库:import torch
2. 创建一个tensor x,可以使用torch.tensor()函数:x = torch.tensor([[1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12]])
3. 使用x.view()方法来修改x的维度为(3, 4):x = x.view(3, 4)
4. 打印修改后的tensor x:print(x)
完整的代码如下:
import torch
x = torch.tensor([[1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12]])
x = x.view(3, 4)
print(x)
上述代码中,我们首先将pytorch库导入,然后创建了一个大小为(2, 6)的tensor x,并使用x.view()方法将其修改为大小为(3, 4)的tensor。最后,我们打印出修改后的tensor x。运行以上代码,输出结果为:
tensor([[ 1, 2, 3, 4],
[ 5, 6, 7, 8],
[ 9, 10, 11, 12]])
可以看到,tensor x的维度已经成功修改为(3, 4)。
阅读全文