flâneur

torch_geometric.data.Data — pytorch_geometric documentation

pytorch-geometric.readthedocs.io · 3,695 words · saved by 1 readers

A data object describing a homogeneous graph. The data object can hold node-level, link-level and graph-level attributes. In general, Data tries to mimic the behavior of a regular Python dictionary. In addition, it provides useful functionality for analyzing graph structures, and provides basic PyTorch tensor functionalities. See here for the accompanying tutorial. x (torch.Tensor, optional) – Node feature matrix with shape [num_nodes, num_node_features]. (default: None) edge_index (LongTensor, optional) – Graph connectivity in COO format with shape [2, num_edges]. (default: None) edge_attr (torch.Tensor, optional) – Edge feature matrix with shape [num_edges, num_edge_features]. (default: None) y (torch.Tensor, optional) – Graph-level or node-level ground-truth labels with arbitrary shape. (default: None) pos (torch.Tensor, optional) – Node position matrix with shape [num_nodes, num_dimensions]. (default: None) time (torch.Tensor, optional) – The timestamps for each event with shape [n

torch_geometric.data.Data - pytorch_geometric documentation torch_geometric.data torch_geometric.data.Data View page source torch_geometric.data.Data  class Data ( x : Optional [ Tensor ] = None , edge_index : Optional [ Tensor ] = None , edge_attr : Optional [ Tensor ] = None , y : Optional [ Union [ Tensor , int , float ] ] = None , pos : Optional [ Tensor ] = None , time : Optional [ Tensor ] = None , ** kwargs ) [source]  Bases: BaseData , FeatureStore , GraphStore A data object describing a homogeneous graph. The data object can hold node-level, link-level and graph-level attributes. In

related reading