m = (Vrf[:, :-1, 1] - s >= Vf[:, 1:, 0]).long()
Vrf[:, 1:, 0] = m * (Vrf[:, :-1, 1] - s) + (1 - m) * Vf[:, 1:, 0]
m = m[:, :, None]
m = (Vrf[:, :-1, 1] - s >= Vf[:, 1:, 0]).long()
Vrf[:, 1:, 0] = m * (Vrf[:, :-1, 1] - s) + (1 - m) * Vf[:, 1:, 0]
m = m[:, :, None]