Inheritance diagram for fastreid.layers.frn.TLU:

Public Member Functions
	__init__ (self, num_features)

	reset_parameters (self)

	extra_repr (self)

	forward (self, x)

Public Attributes
	num_features

	tau

Detailed Description

Definition at line 14 of file frn.py.

Constructor & Destructor Documentation

◆ init()

fastreid.layers.frn.TLU.__init__	(		self,
			num_features )

max(y, tau) = max(y - tau, 0) + tau = ReLU(y - tau) + tau

Definition at line 15 of file frn.py.

    def __init__(self, num_features):
        """max(y, tau) = max(y - tau, 0) + tau = ReLU(y - tau) + tau"""
        super(TLU, self).__init__()
        self.num_features = num_features
        self.tau = Parameter(torch.Tensor(num_features))
        self.reset_parameters()
 

Member Function Documentation

◆ extra_repr()

fastreid.layers.frn.TLU.extra_repr ( self )

Definition at line 25 of file frn.py.

    def extra_repr(self):
        return 'num_features={num_features}'.format(**self.__dict__)
 

◆ forward()

fastreid.layers.frn.TLU.forward	(		self,
			x )

Definition at line 28 of file frn.py.

    def forward(self, x):
        return torch.max(x, self.tau.view(1, self.num_features, 1, 1))

◆ reset_parameters()

fastreid.layers.frn.TLU.reset_parameters ( self )

Definition at line 22 of file frn.py.

    def reset_parameters(self):
        nn.init.zeros_(self.tau)
 

Member Data Documentation

◆ num_features

fastreid.layers.frn.TLU.num_features

Definition at line 18 of file frn.py.

◆ tau

fastreid.layers.frn.TLU.tau

Definition at line 19 of file frn.py.

The documentation for this class was generated from the following file:

smreid/fastreid/layers/frn.py

Public Member Functions

Public Attributes