def __call__(self, src):
    'Augmenter body'
    if (random.random() < self.p):
        src = nd.dot(src, self.mat)
    return src
Augmenter body
def __call__(self, src):
    
    if (random.random() < self.p):
        src = nd.dot(src, self.mat)
    return src