I am looking for equivalent function for numpy.where(condition[, x, y]).
What I am hoping is to have a function for elementwise condition for each element on x & y depends on the boolean matrix “condition”.
But I can’t find it in current pytorch function. Is there any possibility to be included in the future ?