bitorch.layers.debug_layers.WeightGraphicalDebug

class bitorch.layers.debug_layers.WeightGraphicalDebug(module: Module, *args: Any, **kwargs: Any)

Methods

__init__	stores given module
forward	Forwards the input tensor through the debug model and outputs debug information about the given modules weights.

Attributes

__init__(module: Module, *args: Any, **kwargs: Any) → None

stores given module

Parameters:

module (torch.nn.Module) – module the weights of which shall be debugged

forward(x: Tensor) → Tensor

Forwards the input tensor through the debug model and outputs debug information about the given modules weights.

Parameters:

x (torch.Tensor) – the Tensor to be forwarded untouched.

Returns:

the input tensor

Return type:

torch.Tensor