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

edgify / torch python

Repository URL to install this package:

Version: 
2.0.1+cpu  ▾
Details    
torch / nn / quantizable / modules / __init__.py
Size: Mime: 
from torch.ao.nn.quantizable.modules.activation import MultiheadAttention
from torch.ao.nn.quantizable.modules.rnn import LSTM
from torch.ao.nn.quantizable.modules.rnn import LSTMCell

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

Products

About

Resources

Contact Gemfury