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

Before Change

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

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

    if return_transform:

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