Conditionally apply tensor operations in PyTorch

Question

I know PyTorch doesn\'t have a map-like function to apply a function to each element of a tensor. So, could I do something like the following without a map