Hi, i can get a matrix with numpy like this:
import numpy as np
label = np.random.randint(1,6,(2,3))
tmp = np.array(np.where(label == 3)).transpose((1, 0))

if label is array([[1, 2, 3], [3, 3, 4]]).  then tmp will be array([[0, 2], [1, 0], [1, 1]], dtype=int64).
and i want to get the same result for libtorch. here is my code
auto ten = torch::randint(1,10,{2,3});
auto y = torch::full({2,3},3)
auto ind = torch::where(torch::operator==(ten,y), ten, y);

and ind’s data is 3 3 3 3 3 3 [Variable[CPUType]{2,3}]
where’s my fault?