import torch


def shuffle_tensor(input):
    """
    Returns a new tensor whose elements correspond to a randomly shuffled version of the the elements of the input.

    Args:
        input (`torch.Tensor`): input tensor.

    Returns:
        (`torch.Tensor`): output tensor.
    """
    return input[torch.randperm(input.nelement())]