zennit.layer

Additional Utility Layers

Classes

Sum

Compute the sum along an axis.
class zennit.layer.Sum(dim=- 1)[source]

Bases: Module

Compute the sum along an axis.

Parameters

dim (int) – Dimension over which to sum.

forward(input)[source]

Computes the sum along a dimension.