찬호의 Too Much Intelligence
torch.is_tensor() 본문
torch.is_tensor(obj, /)
Returns True if obj is a PyTorch tensor.
내부에서 isinstance(obj, Tensor)를 수행하므로, is_tensor보다 isinstance를 사용하는게 mypy의 type checking 측면에서 더 명확하고 낫다.
따라서 is_tensor보다 isinstance를 사용하자.
x = torch.tensor([1,2,3])
print(torch.is_tensor(x)) # True
# better to use isinstance
print(isinstance(x, torch.Tensor)) # True'PyTorch Reference API > torch' 카테고리의 다른 글
| torch.is_complex() (0) | 2025.10.21 |
|---|---|
| torch.is_storage() (0) | 2025.10.21 |