torch_argmax_mask
=================

.. currentmodule:: pandemonium.policies

.. autofunction:: torch_argmax_mask