zennit.layer

Additional Utility Layers

Classes

Sum	Compute the sum along an axis.

class zennit.layer.Sum(dim=-1)

Bases: Module

Compute the sum along an axis.

Parameters:

dim (int) – Dimension over which to sum.

forward(input)

Computes the sum along a dimension.