Thank you for your reply, @ptrblck.
I want to do some mappings/transferring over the inputs. I saw this https://discuss.pytorch.org/t/multiply-feature-map-by-a-learnable-scalar/1520 40 should use variable and requires_grad? not sure how I can apply that.