def fill_predicted(self, array, predicted, bools):
    bools = bools.cpu().numpy()
    return (((1.0 - bools) * predicted.cpu().numpy()) + array)
def fill_predicted(self, array, predicted, bools):
    bools = bools.cpu().numpy()
    return (((1.0 - bools) * predicted.cpu().numpy()) + array)