Method rev_comsum_dim_2

test_selective_scan_easy.py:690–692

(x)

        delta_softplus = ctx.delta_softplus
        doys = doys.view(B, G, D, L).permute(0, 1, 3, 2)
        def rev_comsum_dim_2(x):
            cum_sum = torch.cumsum(x, dim=2)
            return (x - cum_sum + cum_sum[:,:,-1:None])
            
        dus = None
        dDs = None