pygrad.basics

Module storing class-defined layers.

Classes

AddNorm([gain, bias, epsilon])

Performs AddNorm on an input x and skip connection value skip.

Conv2D(o_dim, i_dim, kH, kW, bias, label[, ...])

Performs Conv2D from an input dimension i_dim to an output dimension o_dim using a kernel (kH, kW).

Dropout([rate])

Dropout Class with specified rate parameter.

Flatten([label])

Flattens an input by reshaping it into a 1D Tensor.

Linear(i_dim, o_dim, bias, label[, dtype])

Linear 2D layer.

Softmax([label])

Performs Softmax on an input.