zennit.layer

Additional Utility Layers

Classes

Sum	Compute the sum along an axis.

class zennit.layer.Sum

Bases: Module

Compute the sum along an axis.

Parameters:

dim (int) – Dimension over which to sum.

__init__(dim=-1)

Initialize internal Module state, shared by both nn.Module and ScriptModule.

forward(input)

Computes the sum along a dimension.

Parameters:

input (torch.Tensor) – The input on which to sum.

Returns:

The resulting tensor summed along dimension dim.

Return type:

torch.Tensor