respect to which to take diagonal. Default: 1.
Note:
 "src" must be of the proper size in order to be embedded into
 "input". Specifically, it should have the same shape as
 "torch.diagonal(input, offset, dim1, dim2)"

Examples:
  >>> a = torch.zeros(3, 3)
  >>> a
  tensor([[0., 0., 0.],
          [0., 0., 0.],
          [0., 0., 0.]])

  >>> torch.diagonal_scatter(a, torch.ones(3), 0)
  tensor([[1., 0., 0.],
          [0., 1., 0.],
          [0., 0., 1.]])

  >>> torch.diagonal_scatter(a, torch.ones(2), 1)
  tensor([[0., 1., 0.],
          [0., 0., 1.],
          [0., 0., 0.]])
