def is_tensor(object_): missed_tensor_classes = {torch.HalfTensor} return torch.is_tensor(object_) or type(object_) in missed_tensor_classes