Convert for loop using broadcasting
I have three tensors X,Y,Z of dimensions (K,) (K,) (n,) and a function k that takes two inputs and returns a float.
I have three tensors X,Y,Z of dimensions (K,) (K,) (n,) and a function k that takes two inputs and returns a float.