pytorch-stateful-lstm 1.6.0

功能

pytorch lstm的实现由libtorch提供支持，并有：

• 隐藏/单元格剪辑。
• 跳过连接。
• 可变辍学和辍学连接。
• 管理的初始状态。
• 内置TBPTT。

基准：https://github.com/cnt-dev/pytorch-stateful-lstm/tree/master/benchmark

安装

先决条件：火炬& gt；= 1.0.0，支持C++ 11编译器（参见here）。通过PIP安装：

pip install pytorch-stateful-lstm

使用量

示例：

import torch
from torch.nn.utils.rnn import pack_padded_sequence, PackedSequence
from pytorch_stateful_lstm import StatefulUnidirectionalLstm

lstm = StatefulUnidirectionalLstm(
        num_layers=2,
        input_size=3,
        hidden_size=5,
        cell_size=7,
)

inputs = pack_padded_sequence(torch.rand(4, 5, 3), [5, 4, 2, 1], batch_first=True)
raw_packed_outputs, lstm_state = lstm(
        inputs.data,
        inputs.batch_sizes
)
outputs = PackedSequence(raw_packed_outputs, inputs.batch_sizes)

有关参数的定义，请参见https://github.com/cnt-dev/pytorch-stateful-lstm/tree/master/extension。

学分

这个包是用Cookiecutter和audreyr/cookiecutter-pypackage项目模板创建的。