pytorch 随即旋转

在 PyTorch 中，可以使用 torchvision.transforms 模块中的 RandomRotation 类来实现随机旋转。下面是一个示例代码：

import torch
from torchvision import transforms

# 创建随机旋转的转换
random_rotation = transforms.RandomRotation(degrees=45)

# 加载图像
image = torch.randn(3, 224, 224)  # 假设图像尺寸为 3x224x224

# 进行随机旋转
rotated_image = random_rotation(image)

# 打印旋转后的图像尺寸
print(rotated_image.size())

在上面的示例中，我们创建了一个 RandomRotation 对象，将 degrees 参数设置为 45，表示图像将随机旋转在 -45 到 +45 度之间。然后，我们将图像传递给 random_rotation 对象来进行随机旋转，返回旋转后的图像。最后，我们打印旋转后的图像尺寸。

请注意，这里的示例使用了随机生成的张量作为图像输入。实际情况中，你需要根据自己的需求加载真实的图像数据。