78d7b1d209c8b5427071d75a9cb8618cf585226b,python/mxnet/ndarray/numpy_extension/_op.py,,masked_softmax,#Any#Any#Any#Any#Any#,137

Before Change

        neg = -1e18
        if _np.dtype(dtype) == _np.float16:
            neg = -1e4
        data = np.where(mask, data, neg)
        logits = (softmax(data, axis=axis) / temperature) * mask
    else:
        logits = softmax(data, axis=axis) / temperature
    return logits

After Change

    array([[0.00669285, 0.        , 0.00669285, 0.        , 0.00669285],
           [0.9933072 , 0.        , 0.9933072 , 0.        , 0.9933072 ]])
    
    assert data is not None and mask is not None, "Missing input data and mask"
    return _api_internal.masked_softmax(data, mask, axis, temperature, normalize)


// pylint: disable=too-many-arguments

Instances