tglite.op.edge_reduce

tglite.op.edge_reduce(blk: TBlock, data: Tensor, op='sum') → Tensor

Computes segmented reduction (e.g. sum or mean) on given data using edge information from the block.

Parameters:

• blk –

• data –