My number of features is 288. I did state.view(1,-1) such that the dimension of the state is now [1, 288] but not I get the following error:

{ValueError}Expected more than 1 value per channel when training, got input size [1, 256]

If i do state.view(-1,1), such that the dimension of the state is [288,1] then I’ll get:

{RuntimeError}size mismatch, m1: [288 x 1], m2: [288 x 256] at c:\programdata\miniconda3\conda-bld\pytorch-cpu_1524541161962\work\aten\src\th\generic/THTensorMath.c:2033