fiftyone.utils.clip.model.AttentionPool2d

class documentation

class AttentionPool2d(nn.Module): (source)

Constructor: AttentionPool2d(spacial_dim, embed_dim, num_heads, output_dim)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method forward Undocumented
Instance Variable c_proj Undocumented
Instance Variable k_proj Undocumented
Instance Variable num_heads Undocumented
Instance Variable positional_embedding Undocumented
Instance Variable q_proj Undocumented
Instance Variable v_proj Undocumented
def __init__(self, spacial_dim: int, embed_dim: int, num_heads: int, output_dim: int = None): (source)

Undocumented

def forward(self, x): (source)

Undocumented

c_proj = (source)

Undocumented

k_proj = (source)

Undocumented

num_heads = (source)

Undocumented

positional_embedding = (source)

Undocumented

q_proj = (source)

Undocumented

v_proj = (source)

Undocumented