Libtorch Tensor indexing using Boolean logic in C++

Question

I am trying to set specific values in a tensor to 0 if they are smaller than a certain threshold. My code looks like this:

#include 
#in