Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Bower components Debian packages RPM packages NuGet packages

edgify / torch   python

Repository URL to install this package:

Version: 2.0.1+cpu 

/ ao / nn / quantizable / modules / __init__.py

from .activation import MultiheadAttention
from .rnn import LSTM
from .rnn import LSTMCell

__all__ = [
    'LSTM',
    'LSTMCell',
    'MultiheadAttention',
]