def METHOD_NAME(self, left_vecs: Tensor, right_vecs: Tensor) -> Tuple[Optional[Tensor], ...]:
    left_vecs = self._expand(left_vecs, self.row_mask)
    right_vecs = self._expand(right_vecs, self.col_mask)
    return self.base.METHOD_NAME(left_vecs, right_vecs) + (None, None)