42418e10a68f534b939a125a9ccd60167f9a6633,torchsample/transforms/torch_transforms.py,AddChannel,__call__,#AddChannel#Any#Any#,141

Before Change

    def __call__(self, x, y=None):
        x = torch.from_numpy(x.numpy().reshape(1,x.size(0), x.size(1)))
        if y is not None:
            y = torch.from_numpy(y.numpy().reshape(1,y.size(0), y.size(1)))
            return x, y
        else:
            return x

After Change

    of size (1, 28, 28), for example.
    
    def __call__(self, x, y=None):
        x = x.unsqueeze(2)
        if y is not None:
            y = y.unsqueeze(2)
            return x,y
        return x

Instances