본문 바로가기

Python/PyTorch

[Pytorch] torch has no "from_numpy" member

#pylint: disableE1101

#pylint: enable=E1101

사이에 torch.from_numpy() 코드를 작성하면 된다.

 

# pylint: disable=E1101
tensor = torch.from_numpy(np_array)
# pylint: enable=E1101