bitorch.layers.debug_layers.InputPrintDebug

class bitorch.layers.debug_layers.InputPrintDebug(debug_interval: int = 100, num_outputs: int = 10, name: str = 'Debug')[source]

Methods

forward

forwards the given tensor without modification, debug output if activated

Attributes

forward(x: Tensor) → Tensor[source]

forwards the given tensor without modification, debug output if activated

Parameters:

x (torch.Tensor) – tensor to be debugged

Returns:

input tensor x

Return type:

torch.Tensor