heat.nn

Neural network submodule.

It contains data parallel specific nn modules. It also includes all of the modules in the torch.nn namespace

Submodules

• heat.nn.data_parallel
• heat.nn.functional

Package Contents

__getattr__(name)

When a function is called for the heat.nn module it will attempt to run the heat nn module with that name, then, if there is no such heat nn module, it will attempt to get the torch nn module of that name.