code: def test_simple_multi_output(self) -> None:
      xs = torch.randn(TestLinearModel.MAX_POINTS, 1)
      y1 = 3 * xs + 1
      y2 = -5 * xs
      ys = torch.stack((y1, y2), dim=1).squeeze()
  
      self.train_and_compare(
          SGDLinearRegression,
          xs,
          ys,
          expected_loss=torch.DoubleTensor([0, 0]),
          expected_reg=torch.DoubleTensor([0, 0]),
          objective="ols",
      )
  
file_path: captum/tests/utils/test_linear_model.py
imports: import torch
  from captum._utils.models.linear_model.model import SGDLasso, SGDLinearRegression, SGDRidge
  from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
label_desc: Use image node_id 3 for calling the TestLinearModel obj's underlying member method code with example usage: obj.test_simple_multi_output() without return types