3ba5bc45ad3ceaafdade9378483388a2d2f49f28,kornia/augmentation/base.py,_BasicAugmentationBase,forward,#_BasicAugmentationBase#Any#Any#,120

Before Change

        in_tensor = self.transform_tensor(in_tensor)
        batch_shape = in_tensor.shape
        if params is None:
            params = self.__forward_parameters__(batch_shape, self.p, self.p_batch, self.same_on_batch)
        self._params = params

        output = self.apply_func(input, self._params)
        return _transform_output_shape(output, ori_shape) if self.keepdim else output

After Change

        in_tensor = self.transform_tensor(in_tensor)
        batch_shape = in_tensor.shape
        if params is None:
            params = self.forward_parameters(batch_shape)
        self._params = params

        output = self.apply_func(input, self._params)
        return _transform_output_shape(output, ori_shape) if self.keepdim else output

Instances