train.AlignLayer Class Reference

Inheritance diagram for train.AlignLayer:

Collaboration diagram for train.AlignLayer:

Public Member Functions
	__init__ (self)
torch.Tensor	forward (self, torch.Tensor batch)

Public Attributes
	shiftV
	shiftZ

Detailed Description

Definition at line 78 of file train.py.

Constructor & Destructor Documentation

__init__()

train.AlignLayer.__init__

(

self

)

Definition at line 79 of file train.py.

References train.AlignLayer.__init__().

Referenced by train.AlignLayer.__init__().

Member Function Documentation

forward()

torch.Tensor train.AlignLayer.forward	(		self,
		torch.Tensor	batch
	)

Definition at line 84 of file train.py.

References train.AlignLayer.shiftV, and train.AlignLayer.shiftZ.

Member Data Documentation

shiftV

train.AlignLayer.shiftV

Definition at line 81 of file train.py.

Referenced by train.AlignLayer.forward().

shiftZ

train.AlignLayer.shiftZ

Definition at line 82 of file train.py.

Referenced by train.AlignLayer.forward().

---

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

• align_new/nn/train.py