neural_tangents.stax.Padding

class neural_tangents.stax.Padding(value)

Type of padding in pooling and convolutional layers.

CIRCULAR

circular padding, as if the input were a torus.

SAME

same, a.k.a. zero padding.

VALID

valid, a.k.a. no padding.

__init__()

Attributes

CIRCULAR
SAME
VALID