Binary masks?

[ryanmrichard]

Maybe I missed it, but is it possible to generate a binary-mask for a tensor? Basically given a tensor T, I want to generate a tensor M which is the same shape as T, with boolean elements such that a particular element of M is true if the corresponding element of T satisfies some condition. I can get the same effect with apply_ewise, but with substantially greater memory cost.

[erdalmutlu]

This can be implemented via a utility function which kind of does a screening over the input tensor and generate a boolean tensor for the corresponding mask. I will keep it as a to-do item if we still have a use-case.