spyrit.core.nnet.Identity

class spyrit.core.nnet.Identity

Bases: Module

Identity layer. Can be useful for ablation study.

This layer returns the input tensor as is.

This class has no arguments and no attributes.

Methods

forward(x)

Forward pass of the Identity layer.