towhee.models.layers.layers_with_relpropΒΆ

Functions

backward_hook

forward_hook

safe_divide

Classes

AdaptiveAvgPool2d

Add

Add with rel prop

AddEye

Add eye

AvgPool2d

BatchNorm2d

BatchNorm2d with rel prop

Cat

Cat with rel prop

Clone

Clone with rel prop

Conv2d

Conv2d with rel prop

Conv3d

Conv3d layer with relprop

Dropout

Einsum

einsum with rel prop

GELU

IndexSelect

IndexSelect with rel prop

LayerNorm

Linear

Linear with rel prop

MaxPool2d

MaxPool3d

ReLU

RelProp

Rel prop basic class

RelPropSimple

Rel prop simple class

Sequential

Sequential with rel prop

Sigmoid

Softmax