gigl.src.common.types.BatchEmbeddings

class gigl.src.common.types.task_inputs.BatchEmbeddings(query_embeddings: torch.FloatTensor, repeated_query_embeddings: Dict[CondensedEdgeType, torch.FloatTensor], pos_embeddings: Dict[CondensedEdgeType, torch.FloatTensor], hard_neg_embeddings: Dict[CondensedEdgeType, torch.FloatTensor], random_neg_embeddings: Dict[CondensedNodeType, torch.FloatTensor])

Bases: object

Methods

__init__

__eq__(other)

Return self==value.

__hash__ = None

__init__(query_embeddings: FloatTensor, repeated_query_embeddings: Dict[CondensedEdgeType, FloatTensor], pos_embeddings: Dict[CondensedEdgeType, FloatTensor], hard_neg_embeddings: Dict[CondensedEdgeType, FloatTensor], random_neg_embeddings: Dict[CondensedNodeType, FloatTensor]) → None

__repr__()

Return repr(self).

__weakref__

list of weak references to the object (if defined)