快捷方式

RNNBase

class torch.nn.RNNBase(mode, input_size, hidden_size, num_layers=1, bias=True, batch_first=False, dropout=0.0, bidirectional=False, proj_size=0, device=None, dtype=None)[source]

RNN 模块(RNN、LSTM、GRU)的基类。

实现 RNN 的各个方面,这些方面在 RNN、LSTM 和 GRU 类中共享,例如模块初始化和用于参数存储管理的实用程序方法。

注意

RNNBase 类未实现 forward 方法。

注意

LSTM 和 GRU 类覆盖了 RNNBase 实现的某些方法。

flatten_parameters()[source]

重置参数数据指针,以便它们可以使用更快的代码路径。

目前,这仅在模块位于 GPU 上且启用了 cuDNN 时有效。 否则,它将是无操作。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得解答

查看资源