찬호의 Too Much Intelligence

torch.is_tensor() 본문

PyTorch Reference API/torch

torch.is_tensor()

frogbam07 2025. 10. 21. 01:15

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