code: def test_forward(self, input):
      actual = kornia.geometry.subpix.spatial_softmax2d(input)
      assert (
          actual.lt(0).sum().item() == 0
      ), "expected no negative values"
      sums = actual.sum(-1).sum(-1)
      assert_close(sums, torch.ones_like(sums))
  
file_path: kornia/test/geometry/subpix/test_dsnt.py
imports: import pytest
  import torch
  import kornia
  from kornia.testing import assert_close
label_desc: Use image node_id 2 for calling the TestSpatialSoftmax2d obj's underlying member method code with example usage: obj.test_forward(input) without return types