No, I used yours. But if i try 16 * 16 * 6 i get the error:
input, target, size_average, reduce)
File "D:\Programme\Anaconda\envs\DeepLearning\lib\site-packages\torch\nn\functional.py", line 1537, in _pointwise_loss
return lambd_optimized(input, target, size_average, reduce)
RuntimeError: input and target shapes do not match: input [10 x 2], target [1 x 3] at c:\programdata\miniconda3\conda-bld\pytorch_1524546371102\work\aten\src\thnn\generic/MSECriterion.c:13