本文简要介绍python语言中 torch.nn.quantized.dynamic.RNNCell
的用法。
用法:
class torch.nn.quantized.dynamic.RNNCell(input_size, hidden_size, bias=True, nonlinearity='tanh', dtype=torch.qint8)
具有 tanh 或 ReLU 非线性的 Elman RNN 单元。以浮点张量作为输入和输出的动态量化 RNNCell 模块。权重被量化为 8 位。我们采用与
torch.nn.RNNCell
相同的接口,请参阅https://pytorch.org/docs/stable/nn.html#torch.nn.RNNCell文档。例子:
>>> rnn = nn.RNNCell(10, 20) >>> input = torch.randn(6, 3, 10) >>> hx = torch.randn(3, 20) >>> output = [] >>> for i in range(6): hx = rnn(input[i], hx) output.append(hx)
相关用法
- Python PyTorch RNNCell用法及代码示例
- Python PyTorch RNNTLoss用法及代码示例
- Python PyTorch RNN用法及代码示例
- Python PyTorch RRef.rpc_sync用法及代码示例
- Python PyTorch RReLU用法及代码示例
- Python PyTorch ReduceLROnPlateau用法及代码示例
- Python PyTorch ReflectionPad1d用法及代码示例
- Python PyTorch RRef.backward用法及代码示例
- Python PyTorch RemoteModule用法及代码示例
- Python PyTorch Rows2Columnar用法及代码示例
- Python PyTorch RandomRecDataset用法及代码示例
- Python PyTorch ReplicationPad1d用法及代码示例
- Python PyTorch Resample用法及代码示例
- Python PyTorch ReplicationPad3d用法及代码示例
- Python PyTorch ReflectionPad2d用法及代码示例
- Python PyTorch RandomApply用法及代码示例
- Python PyTorch RRef.remote用法及代码示例
- Python PyTorch RarArchiveLoader用法及代码示例
- Python PyTorch RelaxedOneHotCategorical用法及代码示例
- Python PyTorch RandomErasing用法及代码示例
- Python PyTorch RendezvousHandler.shutdown用法及代码示例
- Python PyTorch ReLU用法及代码示例
- Python PyTorch ReLU6用法及代码示例
- Python PyTorch ReplicationPad2d用法及代码示例
- Python PyTorch ReflectionPad3d用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.quantized.dynamic.RNNCell。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。