docs: def _create_dataset(self, *data):
          """Converts input data to the appropriate Dataset"""
          # Make sure data is a tuple of dense tensors
          data = [self._to_torch(x, dtype=torch.FloatTensor) for x in data]
          return TensorDataset(*data)