torch rotate
时间: 2024-07-06 18:00:36 浏览: 172
torch-1.0.0
5星 · 资源好评率100%
`torch.rotate` 是 PyTorch 中的一个功能,用于在给定的张量上执行旋转操作。它主要用于处理一维或二维数据,比如图像或向量,可以按照指定的角度顺时针或逆时针旋转。这个函数可以接受一个张量(Tensor)和一个旋转角度(angle),通常以弧度为单位。以下是一个简单的使用示例:
```python
import torch
tensor = torch.tensor([[1, 2, 3], [4, 5, 6]])
rotated_tensor = torch.rotate(tensor, angle=math.pi/2) # 90度顺时针旋转
```
阅读全文