dlutils.pytorch package

Submodules

dlutils.pytorch.cuda_helper module

dlutils.pytorch.jacobian module

dlutils.pytorch.jacobian.jacobian(inputs, output)

Parameters

• inputs – Batch X Size (e.g. Depth X Width X Height)

• output – Batch X Classes

Returns

jacobian: Batch X Classes X Size

Module contents