错误(未解决)——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。解决方法