Batch

class marius.data.Batch

to(self: marius._data.Batch, device: torch.device) → None

accumulateGradients(self: marius._data.Batch, learning_rate: float) → None

property batch_id

property batch_size

clear(self: marius._data.Batch) → None

property dense_graph

property device_id

property dst_neg_filter

property dst_neg_indices

property dst_neg_indices_mapping

property edges

embeddingsToHost(self: marius._data.Batch) → None

property encoded_uniques

property neg_edges

property node_embeddings

property node_embeddings_state

property node_features

property node_gradients

property node_labels

property node_state_update

property rel_neg_indices

property root_node_indices

property src_neg_filter

property src_neg_indices

property src_neg_indices_mapping

property start_idx

property status

property train

property unique_node_indices