What is the most convenient way to calculate the row-wise abs value of two numbers across the first dimension of the tensor, i.e. the result should be a n-1 vector with the entries equal to sqrt(a^2 + b^2) for every row. Thanks in adance