neodroidvision.mixed.architectures.self_attention_network.self_attention_network.make_san

neodroidvision.mixed.architectures.self_attention_network.self_attention_network.make_san(*, self_attention_type: SelfAttentionTypeEnum = SelfAttentionTypeEnum.pairwise, layers: Sequence, kernels: Sequence, num_classes: int) → Module[source]
Parameters
  • self_attention_type –

  • layers –

  • kernels –

  • num_classes –

Returns

Return type