Link Search Menu Expand Document

API Reference

The Torch-HD API is pythonic and follows the structure of PyTorch. The API consists of 2 components. A hdlayers module which implements the Encoders (as PyTorch custom layers) and a utitlities module which implements helper functions such as trainer etc.


Table of contents