记录pytorch代码中的运行报错:
RuntimeError: size mismatch, m1: [1 x 100], m2: [1 x 1] at c:\programdata\miniconda3\conda-bld\pytorch-cpu_1532498166916\work\aten\src\th\generic/THTensorMath.cpp:2070
解决方法:
在此处torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)加上dim=1