@alexis-jacq I need to access the intermediate data… Can I do that in your implementation?
@apaszke I thought it would work too, but it says
matrices expected, got 4D, 2D tensors at /data/users/soumith/miniconda2/conda-bld/pytorch-0.1.7_1485444530918/work/torch/lib/TH/generic/THTensorMath.c:857