Yes, I’ve already understood it from other similar issues. It worked in this way with pytorch==0.4.0, I guess the meaning of this was “derive results w.r.t each element in the vector”, so to remove it will do the work, don’t you think @albanD ?
At least after trying it, I see it gives me same results as in the old version, which I did manage to run on google Collab.