torch_geometric.utils — pytorch_geometric documentation
Reduces all values from the src tensor at the indices specified in the index tensor along a given dimension dim. group_argsort Returns the indices that sort the tensor src along a given dimension in ascending order by value. segment Reduces all values in the first dimension of the src tensor within the ranges specified in the ptr. index_sort Sorts the elements of the inputs tensor in ascending order. cumsum Returns the cumulative sum of elements of x. degree Computes the (unweighted) degree of a given one-dimensional index tensor. softmax Computes a sparsely evaluated softmax. lexsort Performs an indirect stable sort using a sequence of keys. sort_edge_index Row-wise sorts edge_index. coalesce Row-wise sorts edge_index and removes its duplicated entries. is_undirected Returns True if the graph given by edge_index is undirected. to_undirected Converts the graph given by edge_index to an undirected graph such that ( 𝑗 , 𝑖 ) ∈ 𝐸 for every edge ( 𝑖 , 𝑗 ) ∈ 𝐸 . contains_self_loops
torch_geometric.utils - pytorch_geometric documentation --> torch_geometric.utils Edit on GitHub torch_geometric.utils scatter Reduces all values from the src tensor at the indices specified in the index tensor along a given dimension dim . group_argsort Returns the indices that sort the tensor src along a given dimension in ascending order by value. segment Reduces all values in the first dimension of the src tensor within the ranges specified in the ptr . index_sort Sorts the elements of the inputs tensor in ascending order. cumsum Returns the cumulative sum of elements of x . degree Compu
Explore this link on the map →related reading
- torch_geometric.data.Data - pytorch_geometric documentationpytorch-geometric.readthedocs.io
- torch.index_select — PyTorch 2.12 documentationdocs.pytorch.org
- A Gentle Introduction to Graph Neural Networksdistill.pub
- PyTorch internals : ezyang's blogblog.ezyang.com
- torch.Tensor — PyTorch 2.12 documentationdocs.pytorch.org
- torch.max — PyTorch 2.12 documentationdocs.pytorch.org
- torch.where — PyTorch 2.12 documentationdocs.pytorch.org
- Chapter 0: Fundamentals - ARENAlearn.arena.education
- torch.triu — PyTorch 2.12 documentationdocs.pytorch.org
- Tensors — PyTorch Tutorials 2.12.0+cu130 documentationpytorch.org
- Understanding Convolutions on Graphsdistill.pub
- python - How to convert a PyTorch sparse_coo_tensor into a PyTorch dense tensor? - Stack Overflowstackoverflow.com