티스토리 뷰

Computer/이게 왜 안되지?

[이왜안?] TypeError: expected np.ndarray (got Tensor)

벼랑끝과학자 2022. 12. 31. 03:14

brief solution

1. torch.from_numpy needs np.ndarray arguments not an Tensor

 

인자로 numpy의 ndarray를 받아야 하는 from_numpy에 실수로 Tensor를 전달하고 있었다.

        label = Variable(torch.from_numpy(label).float()).to(device)

해당코드를 아래와 같이 수정하면 해결할 수 있다.

        label = Variable(torch.from_numpy(np.array(label)).float()).to(device)

본인 코드에서 인자로 ndarray를 받아야 하는 함수에 잘못된 Tensor 타입의 데이터를 전달하고 있는지를 확인해보자

저작자표시

'Computer > 이게 왜 안되지?' 카테고리의 다른 글

[이왜안?] nn.Embedding, IndexError: index out of range in self  (0) 2023.02.13
[이왜안?] ImportError: /usr/lib/x86_64-linux-gnu/libstdc++.so.6: version `GLIBCXX_3.4.29' not found (required by /home/../anaconda3/envs/DTA/lib/python3.9/site-packages/pandas/_libs/window/aggregations.cpython-39-x86_64-linux-gnu.so)  (0) 2023.02.03
[이왜안?] Input and hidden tensors are not at the same device, found input tensor at cuda:0 and hidden tensor at cpu  (0) 2022.12.31
[이왜안?] Input type (torch.FloatTensor) and weight type (torch.DoubleTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor  (0) 2022.12.30
[이왜안?] try, except문 내부에서는 함수 오타를 조심하자!  (0) 2022.12.22
댓글