jittor_geometric.ops

Author: lusz Date: 2024-06-21 11:01:25 Description:

jittor_geometric.ops.EdgeSoftmax(x, csc)

jittor_geometric.ops.ScatterToEdge(x, csc, flow)

jittor_geometric.ops.ScatterToVertex(x, csc, flow)

jittor_geometric.ops.SpmmCoo(x, edge_index, edge_weight, trans_A=True, trans_B=False)

jittor_geometric.ops.SpmmCsr(x, csr, trans_A=True, trans_B=False)

jittor_geometric.ops.aggregateWithWeight(x, csc, csr, weight=None)

jittor_geometric.ops.cootocsc(edge_index, edge_weight, v_num)

jittor_geometric.ops.cootocsr(edge_index, edge_weight, v_num)

jittor_geometric.ops.from_nodes(csc, nodes)

Given a CSC structure and a set of input nodes, find all the neighbor nodes.

Parameters:

• csc (CSC) – Compressed Sparse Column structure.

• nodes (Var) – Input node IDs (Var type).

Returns:

A Var containing all neighbor nodes corresponding to the input nodes.

Return type:

Var

jittor_geometric.ops.toUndirected(edge_index, edge_attr, num_nodes)

jittor_geometric.ops.to_nodes(csr, nodes)

Given a CSR structure and a set of input nodes, find all the neighbor nodes.

Parameters:

• csr (CSR) – Compressed Sparse Row structure.

• nodes (Var) – Input node IDs (Var type).

Returns:

A Var containing all neighbor nodes corresponding to the input nodes.

Return type:

Var

```