ca42233dc8a72cf8440183d34f542f51c063d5fe,gpytorch/lazy/lazy_variable.py,LazyVariable,transpose,#LazyVariable#Any#Any#,228

Before Change

        // Batch case
        if dim1 < ndimension - 2 and dim2 < ndimension - 2:
            res = self.__class__(self.base_lazy_variable.transpose(dim1, dim2),
                                 self.left_interp_indices.transpose(dim1, dim2),
                                 self.left_interp_values.tranpose(dim1, dim2),
                                 self.right_interp_indices.transpose(dim1, dim2),
                                 self.right_interp_values.transpose(dim1, dim2))

        elif dim1 >= ndimension - 2 and dim2 >= ndimension - 2:
            res = self._transpose_nonbatch()

After Change

        // Batch case
        if dim1 < ndimension - 2 and dim2 < ndimension - 2:
            res = self.__class__(*(arg.transpose(dim1, dim2) for arg in self._args))

        elif dim1 >= ndimension - 2 and dim2 >= ndimension - 2:
            res = self._transpose_nonbatch()

Instances