给我一个使用unsqueeze(-1)的例子
时间: 2024-05-03 18:16:34 浏览: 52
使用python编写一个程序,采用神经网络的方式解决二分类问题
假设有一个形状为 (3, 4) 的张量 tensor,我们想要在最后一个维度上增加一个新的维度,可以使用 unsqueeze(-1) 方法来实现:
```python
import torch
tensor = torch.randn(3, 4)
print(tensor.shape) # 输出:torch.Size([3, 4])
new_tensor = tensor.unsqueeze(-1)
print(new_tensor.shape) # 输出:torch.Size([3, 4, 1])
```
在上面的例子中,我们传入了 -1 参数,表示在最后一个维度上增加新的维度。这样,原本形状为 (3, 4) 的张量变成了形状为 (3, 4, 1) 的张量。可以看到,新的张量在最后一个维度上新增了一个维度。
阅读全文