braindecode.modules.CAT#

class braindecode.modules.CAT(in_channels: int, reduction_rate: int, kernel_size: int, bias=False)[source]#

Attention Mechanism from [Wu2023].

Parameters:

• in_channels (int) – number of input feature channels

• reduction_rate (int) – reduction ratio of the fully-connected layers

• kernel_size (int) – kernel size of the convolutional layer

• bias (bool, default=False) – if True, adds a learnable bias will be used in the convolution,

References

[Wu2023]

Wu, Z. et al., 2023 CAT: Learning to Collaborate Channel and Spatial Attention from Multi-Information Fusion. IET Computer Vision 2023.

Methods

forward(x)[source]#

Apply the CAT block to the input tensor.

Parameters:

x (Pytorch.Tensor)

Return type:

Pytorch.Tensor