Pytorch錯誤集錦

錯誤（未解決）——RuntimeError: dot: Expected 1-D argument self, but got 2-D

import torch
import numpy as np

#abs
data = [[1,-1],[2,-3]]
data = np.array(data)
tensor = torch.FloatTensor(data)

print("\ntensor:",tensor)
print("\n",
      "\nnumpy:",data.dot(data),
      "\ntorch:",tensor.dot(tensor)
      )

錯誤：RuntimeError: CUDA error: out of memory

這裏的out of memory指的是在參數調整的很小的時候依然有問題，比如batch_size=1。解決方法