Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 121 Bytes
Newer Older
Miltos Allamanis's avatar
A first round of implementation of scatter_logsumexp/softmax/logsoftmax ops.  
Miltos Allamanis committed
1
2
3
4
5
6 
from .softmax import scatter_log_softmax, scatter_softmax

__all__ = [
    'scatter_softmax',
    'scatter_log_softmax'
]