EdgeDecoder

class marius.nn.decoders.edge.EdgeDecoder

__init__()

apply_relation(self: marius._nn.decoders.edge.EdgeDecoder, nodes: torch.Tensor, relations: torch.Tensor) → torch.Tensor

compute_scores(self: marius._nn.decoders.edge.EdgeDecoder, src: torch.Tensor, dst: torch.Tensor) → torch.Tensor

select_relations(self: marius._nn.decoders.edge.EdgeDecoder, indices: torch.Tensor, inverse: bool = False) → torch.Tensor

property comparator

property embedding_size

property inverse_relations

property mode

property num_relations

property relation_operator

property relations

property tensor_options

property use_inverse_relations