title: Replace all nonzero values by zero and all zero values by a specific value
body: <p>I have a 3d tensor which contains some zero and nonzero values. I want to replace all nonzero values by zero and zero values by a specific value. How can I do that?</p>