def dummy_tensor_dataset():
    """Returns a PyTorch image dataset of 10 classes."""
    x = torch.rand(32, 10)
    y = torch.rand(32, 10)
    return TensorDataset(x, y)
Returns a PyTorch image dataset of 10 classes.
tests/unit_tests_utils.py