What is np.where's equivalent in libtorch?

C++
1

I can get the index of matrix elements where the values hold particular condition, by:

import numpy as np 
matrix = np.random.randint(0,8,(4,4)) 
inds =np.where(matrix<5)

How do I achieve the same result using torch::where(.) ?
It’s seem that torch::where(.) has different api with np.where(.) .

2

I installed the excellent NumCpp library, which is a header only implementation of numpy for C++. Works fine for me.