def METHOD_NAME(request):
    return [
        (torch.empty(10).uniform_(0, 10), torch.empty(10).uniform_(0, 10), 1),
        (torch.empty(10, 1).uniform_(-10, 10), torch.empty(10, 1).uniform_(-10, 10), 1),
        # updated batches
        (torch.empty(50).uniform_(0, 10), torch.empty(50).uniform_(0, 10), 16),
        (torch.empty(50, 1).uniform_(-10, 10), torch.empty(50, 1).uniform_(-10, 10), 16),
    ][request.param]