Pytorch_Ext

Extension to Pytorch, including new modules, new functionals, bug fixes, etc.

Install

Install from source to get the up-to-date version:

pip install git+https://github.com/david-leon/Pytorch_Ext.git

Python Module	Explanation
module	all neual network module definitions
functional	operations on tensor with no parameter to be learned
objective	objective definitions
util	utility functions