You could convert your tensors into numpy arrays and use scikit learn’s metric functions. That should be pretty straight forward. However, we could actually try writing native metric functions in pytorch.