ab9c4f8474ca744c7e67822fa21d408d0b0ae2d0,kornia/augmentation/functional.py,,apply_rotation,#Any#Any#Any#,469

Before Change

                                      wont be concatenated
    

    if not torch.is_tensor(input):
        raise TypeError(f"Input type is not a torch.Tensor. Got {type(input)}")

    device: torch.device = input.device
    dtype: torch.dtype = input.dtype

    input = input.unsqueeze(0)
    input = input.view((-1, (*input.shape[-3:])))
    angles: torch.Tensor = params["degrees"].to(device, dtype)

    transformed: torch.Tensor = rotate(input, angles).squeeze(0)

After Change

    angles: torch.Tensor = params["degrees"].type_as(input)

    transformed: torch.Tensor = rotate(
        input, angles, mode=Resample(params["interpolation"].item()).name.lower())

    if return_transform:
        // TODO: This part should be inferred from rotate directly

Instances