Skip to main content

pytorch tree lstm package

Project description


This repository contains a Pytorch Implementation of "Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networks " (

This contains two type of tree-lstm (Child sum, N-ary). This was tested by Python 3.6, Pytorch 1.3.0., and this internally uses dgl 0.4.0

This repository referenced


pip install tree-lstm

after installed, you can use this via

import TreeLSTM, Tree, BatchedTree


  1. make Tree object (initialize with arbitrary tensor)
  2. make BatchedTree object using list of Tree object
  3. make TreeLSTM object (inherited from torch.nn.Module)
  4. pass BatchedTree object into TreeLSTM object

For more detailed usage, please see

Thanks for reporting issues / contributions / starts / watches!

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tree_lstm-0.0.8.tar.gz (3.0 kB view hashes)

Uploaded source

Built Distribution

tree_lstm-0.0.8-py3-none-any.whl (4.3 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page