cogdl.layers.se_layer

Module Contents

Classes

SELayer

Squeeze-and-excitation networks

class cogdl.layers.se_layer.SELayer(in_channels, se_channels)[source]

Bases: torch.nn.Module

Squeeze-and-excitation networks

forward(self, x)[source]