def METHOD_NAME(self, column):
    import torch
    if isinstance(column, list) and column:
        if all(
            isinstance(x, torch.Tensor) and x.shape == column[0].shape and x.dtype == column[0].dtype
            for x in column
        ):
            return torch.stack(column)
    return column