I have a matrix where I want to transform each row element-wise with a different function. It’s not a simple linear function, but a non-linear one. Is this possible in pytorch?
My set of functions is not that small (it depends on the dimensionality of the input).