Skip to main content

pytorch-stateful-lstm

https://img.shields.io/pypi/v/pytorch_stateful_lstm.svg https://img.shields.io/travis/cnt-dev/pytorch-stateful-lstm.svg
  • Free software: MIT license

Features

Pytorch LSTM implementation powered by Libtorch, and with the support of:

  • Hidden/Cell Clip.

  • Skip Connections.

  • Variational Dropout & DropConnect.

  • Managed Initial State.

  • Built-in TBPTT.

Benchmark: https://github.com/cnt-dev/pytorch-stateful-lstm/tree/master/benchmark

Install

Prerequisite: torch>=1.0.0, supported C++11 compiler (see here). To install through pip:

pip install pytorch-stateful-lstm

Usage

Example:

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)

For the definition of parameters, see https://github.com/cnt-dev/pytorch-stateful-lstm/tree/master/extension.

Credits

This package was created with Cookiecutter and the audreyr/cookiecutter-pypackage project template.

History

0.1.0 (2019-01-03)

  • First release on PyPI.

Metadata

Release files for pytorch-stateful-lstm 1.6.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for pytorch-stateful-lstm 1.6.0
File Size Uploaded Provenance
pytorch_stateful_lstm-1.6.0.tar.gz 18.6 kB Without provenance Details

Release files / pytorch_stateful_lstm-1.6.0.tar.gz

Download URL pytorch_stateful_lstm-1.6.0.tar.gz
Size 18.6 kB
Tags Source
SHA-256 checksum
How to use checksums
13b23fd6d68198489f7ded7e413ae99ca770a2e27ec1b0fcd4347f0f0408d4d4
BLAKE2b-256 checksum
How to use checksums
2698684219d6b443dbb1cbbcea88859821a1126d4ac2b75798a9869f39eecbf5
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/1.13.0 pkginfo/1.5.0.1 requests/2.21.0 setuptools/40.8.0 requests-toolbelt/0.9.1 tqdm/4.31.1 CPython/3.6.7

Security for pytorch-stateful-lstm 1.6.0

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page