scatter_edge_to_vert_vert

iskra.topology.scatter_edge_to_vert_vert(values: Tensor) → Tensor[SOURCE]

Scatters scalar edge data to vertex-vertex data.

Under the hood, this simply duplicates scalar edge data so it becomes scalar vert-vert data:

return torch.cat([values, values], -1)

See also

iskra.topology.get_vert_vert().

Parameters:

values (Tensor[DType, [Bs, E]]) – Scalar per-edge data.

Returns:

​ (Tensor[DType, [Bs, 2:abbr:E (Number of edges)]]) – Duplicated edge data.