__init__.py 122 Bytes
Newer Older
1
2
3
4
5
from .softmax import scatter_log_softmax, scatter_softmax

__all__ = [
    'scatter_softmax',
    'scatter_log_softmax'
Miltos Allamanis's avatar
Miltos Allamanis committed
6
]