sample is no a numpy array as i had to change it to split the image. Is there a quick method to change it back please?
    sample[:, :, 0].zero_()
AttributeError: 'numpy.ndarray' object has no attribute 'zero_'


Thanks