SelfAttn2D

class aixd.mlmodel.architecture.blocks.SelfAttn2D(*args: Any, **kwargs: Any)[source]

Bases: SelfAttn

Module implementing a 2D self-attention layer.

Parameters:
  • in_channels (int) – The number of input channels.

  • eps (float, optional, default=1e-12) – The epsilon value for the spectral normalization.

Inherited Methods