作为训练卷积神经网络的一部分,我正在使用一个包含
<class 'torch.Tensor'>
对象的数组inputs
。我想将单个<class 'torch.Tensor'>
对象旋转一些随机角度x
,如下所示:def rotate(inputs, x):
# Rotate inputs[0] by x degrees, x can take on any value from 0 - 180 degrees
我该如何做?对于现有的实现,我只能找到torch
有一个rot90
函数,但这限制了我只能旋转90
度的倍数,这并没有帮助我的场景。
谢谢,Vinny
torchvision.transforms.RandomRotation
,但它只适用于PIL图像。我不知道是否有广泛的实现适用于torch.Tensor
对象。 - Szymon Maszke