import torch


def get_test_tensor(max_value=10, float=False):
    """Create simple test tensor."""
    tensor = torch.LongTensor(list(range(max_value)))
    if float:
        tensor = tensor.float()
    return tensor