How can I do the operation the same as `np.where`?

[Pete_Florence]

@truenicoco, torch.where() looks like it’'s in 0.4! http://pytorch.org/docs/master/torch.html#torch.where

thanks others for examples of other options for 0.3 and before