Temp

[Torch.tensor] ValueError: At least one stride in the given numpy array is negative, and tensors with negative strides are not currently supported.

ju_young 2021. 11. 30. 14:55
728x90

다음 코드에서 numpy.ndarray를 torch.tensor로 변환하는 중 에러가 발생하였다.

last_layer.weight.data = torch.tensor(last).unsqueeze(-1).unsqueeze(-1)

다음과 같이 torch.from_numpy()와 copy()를 사용하여 해결하였다.

last_layer.weight.data = torch.from_numpy(last.copy()).unsqueeze(-1).unsqueeze(-1)

 

[ref]

https://stackoverflow.com/questions/57517740/pytorch-custom-dataset-valueerror-some-of-the-strides-of-a-given-numpy-array-a

728x90