import torch


def jitter_soma_depth(feats, scale=10):
    """"
    Apply jitter to soma depth.
    Args:
        feats: features per node
        scale: scale factor of jittering
    """
    new_feats = feats.copy()
    new_feats[:, 1] += torch.randn(1).numpy() * scale
    return new_feats