Function finalize_tile_workspace

mamba_ssm/utils/determinism.py:91–96

(tensor, deterministic)

def finalize_tile_workspace(tensor, deterministic):
    if tensor is None:
        return None
    if deterministic:
        tensor = tensor.sum(dim=-1)
    return tensor