torch-layers

Custom PyTorch layers to add useful features


Keywords
deep-learning, pytorch, pytorch-implementation
License
BSD-1-Clause
Install
pip install torch-layers==0.1.2

Documentation

Custom PyTorch layers

Build Status codecov

This repository implements various layers (either to replace deprecated layers of torch or to add useful new features or to encapsulate parts of the functional API into layers).

Currently the following layers are implemented:

Installation

This package can be installed via pip with:

pip install torch_layers

or from source via:

pip install git+https://github.com/justusschock/torch_layers