bitorch.layers.debug_layers.ShapePrintDebug

class bitorch.layers.debug_layers.ShapePrintDebug(debug_interval: int = 100, num_outputs: int = 10, name: str = 'Debug')

Methods

forward

prints the shape of x, leaves x untouched

Attributes

forward(x: Tensor) → Tensor

prints the shape of x, leaves x untouched

Parameters:

x (torch.Tensor) – the tensor to be debugged

Returns:

input tensor x

Return type:

torch.Tensor