from typing import Any import torch from torchvision.datapoints._datapoint import Datapoint def is_simple_tensor(inpt: Any) -> bool: return isinstance(inpt, torch.Tensor) and not isinstance(inpt, Datapoint)