Hi @Martin1,
I wrote a function here that automates this type of element-wise operation for functions of arbitrary shape so that you don’t have to type all the components of the tensor. Hope you find this helpful.