deeplay.ops.attention.self Module# Classes# MultiheadCrossAttention(*args, **kwargs) MultiheadSelfAttention(*args, **kwargs)