파이토치 tensor 내 nan 값 개수 세아리기
파이토치 tensor 내 nan 값 개수 세아리기 torch 활용 import torch # 예시 텐서 생성 tensor_ex = torch.tensor([1, np.nan, 2, np.nan]) print(tensor_ex) # out : tensor([1., nan, 2., nan]) # nan 값을 True 로 리턴하는 isnan 메서드 활용 mask = torch.isnan(tensor_ex) print(mask) # out : tensor([False, True, False, True]) # True 를 1, False 를 0 으로 변환하는 int() 메서드 활용 mask_int= mask.int() print(mask_int) # out : tensor([0, 1, 0, 1], dtype=to..
2023. 7. 21.