def __call__(self, x, m):
        x = x*m.dimshuffle(0, 'x', 1, 'x')
        e_x = T.exp(x.max(axis=2, keepdims=True) - x)
        e_x = e_x*m.dimshuffle(0, 'x', 1, 'x')
        return e_x / e_x.sum(axis=2, keepdims=True)
        # return e_x / T.clip(e_x.sum(axis=2, keepdims=True), 1e-6, 1e6)
dataset/ETHPy150Open IndicoDataSolutions/Foxhound/foxhound/activations.py/MaskedConvSoftmax.__call__/swapped operands of `-`