用法:
class torch.nn.quantized.dynamic.LSTM(*args, **kwargs)
以浮点张量作为输入和输出的动态量化 LSTM 模块。我们采用相同的接口
torch.nn.LSTM
, 请参见https://pytorch.org/docs/stable/nn.html#torch.nn.LSTM用于文档。例子:
>>> rnn = nn.LSTM(10, 20, 2) >>> input = torch.randn(5, 3, 10) >>> h0 = torch.randn(2, 3, 20) >>> c0 = torch.randn(2, 3, 20) >>> output, (hn, cn) = rnn(input, (h0, c0))
相关用法
- Python torch.nn.quantized.dynamic.LSTMCell用法及代码示例
- Python torch.nn.quantized.dynamic.Linear用法及代码示例
- Python torch.nn.quantized.dynamic.RNNCell用法及代码示例
- Python torch.nn.quantized.dynamic.GRUCell用法及代码示例
- Python torch.nn.quantized.functional.conv2d用法及代码示例
- Python torch.nn.quantized.Conv2d用法及代码示例
- Python torch.nn.quantized.QFunctional用法及代码示例
- Python torch.nn.quantized.Embedding用法及代码示例
- Python torch.nn.quantized.functional.conv3d用法及代码示例
- Python torch.nn.quantized.EmbeddingBag用法及代码示例
- Python torch.nn.quantized.ReLU6用法及代码示例
- Python torch.nn.quantized.Linear用法及代码示例
- Python torch.nn.quantized.functional.conv1d用法及代码示例
- Python torch.nn.quantized.Quantize用法及代码示例
- Python torch.nn.quantized.Conv1d用法及代码示例
- Python torch.nn.quantized.FloatFunctional用法及代码示例
- Python torch.nn.quantized.DeQuantize用法及代码示例
- Python torch.nn.quantized.Conv3d用法及代码示例
- Python torch.nn.InstanceNorm3d用法及代码示例
- Python torch.nn.EmbeddingBag用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.quantized.dynamic.LSTM。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。