_get_unit_square_intercepts

gistfile1.txt

    def _get_unit_square_intercepts(self, slopes, intercept):
        """
        returns unit square intercepts for given slope (a) and intercepts (b)
        y = ax + b
        solves:
        right: y = a + b
        x = 1
        y = slopes + intercept
        left: y = b
        x = 0
        y = intercept
        top: 1 = ax + b
        x = torch.divide(1 - intercept, slopes)
        y = 1
        bottom: 0 = ax + b
        x = torch.divide(- intercept, slopes)
        y = 0

        :param slopes: b x 1
        :param intercepts: b x 1
        :return: points where line intersects unit square borders: b x pts(x, y): b x 2 x 2
        """
        batches = slopes.size(0)
        x = torch.column_stack([torch.ones(batches),
                                torch.zeros(batches),
                                torch.divide(1 - intercept, slopes),
                                torch.divide(-1 * intercept, slopes)])

        y = torch.column_stack([slopes + intercept,
                                intercept,
                                torch.ones(batches),
                                torch.zeros(batches)])

        acceptance = (y >= 0) * (y <= 1) * (x >= 0) * (x <= 1)
        return torch.column_stack((x[acceptance].reshape(batches, 1, -1),
                                   y[acceptance].reshape(batches, 1, -1)))  # b x pts(x, y)