tglite.op.edge_softmax

tglite.op.edge_softmax(blk: TBlock, data: Tensor) → Tensor

Computes segmented softmax on given data using edge information from the block.

Parameters:

• blk –

• data –