파이토치 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=torch.int32)


# tensor 내 정수 합 도출
nan_count = maske_inf.sum().item()
print(nan_count)

# out : 2

 

넘파이 array 활용

import torch
import numpy as np

# 예시 텐서 생성
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])


# 넘파이 array 전환
array_ex = mask.numpy()
# print(array_ex)

# out : [False  True False  True]


# True 의 합 산출
nan_count = np.sum(array_ex == True)
print(nan_count)

# out : 2