gcn_utils.TCNBlock Class Reference

Inheritance diagram for gcn_utils.TCNBlock:

Public Member Functions
	__init__ (self, in_channels, kernel_size, stride, dropout=0)
	forward (self, x)

Public Attributes
	kernel_size
	tcn

Detailed Description

Definition at line 66 of file gcn_utils.py.

Constructor & Destructor Documentation

__init__()

gcn_utils.TCNBlock.__init__	(		self,
			in_channels,
			kernel_size,
			stride,
			dropout = 0 )

Definition at line 68 of file gcn_utils.py.

                 dropout=0):
        super().__init__()
        padding = ((kernel_size - 1) // 2, 0)
        self.kernel_size = kernel_size
        self.tcn = nn.Sequential(
            nn.BatchNorm2d(in_channels),
            nn.ReLU(inplace=True),
            nn.Conv2d(
                in_channels,
                in_channels,
                (kernel_size, 1),
                (stride, 1),
                padding,
            ),
            nn.BatchNorm2d(in_channels),
            nn.Dropout(dropout, inplace=True),
        )
 

Member Function Documentation

forward()

gcn_utils.TCNBlock.forward	(		self,
			x )

Definition at line 90 of file gcn_utils.py.

    def forward(self, x):        
        x = self.tcn(x)
        return x
    

Member Data Documentation

kernel_size

gcn_utils.TCNBlock.kernel_size

Definition at line 75 of file gcn_utils.py.

tcn

gcn_utils.TCNBlock.tcn

Definition at line 76 of file gcn_utils.py.

---

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

• smaction/utils/gcn_utils.py