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