class_docstr: Module that calculates soft argmax 3d per window.
  
  See
  :func: `~kornia.geometry.subpix.conv_soft_argmax3d` for details.
code: def forward(self, x: Tensor) -> Tensor | tuple[Tensor, Tensor]:
      return conv_soft_argmax3d(
          x,
          self.kernel_size,
          self.stride,
          self.padding,
          self.temperature,
          self.normalized_coordinates,
          self.eps,
          self.output_value,
          self.strict_maxima_bonus,
      )
  
imports: from __future__ import annotations
  import torch
  import torch.nn.functional
  from kornia.core import Module, Tensor, concatenate, rand, stack, tensor, where, zeros, zeros_like
  from kornia.filters.sobel import spatial_gradient3d
  from kornia.geometry.conversions import normalize_pixel_coordinates, normalize_pixel_coordinates3d
  from kornia.utils import create_meshgrid, create_meshgrid3d
  from kornia.utils._compat import torch_version_ge
  from kornia.utils.helpers import safe_solve_with_mask
  from .dsnt import spatial_expectation2d, spatial_softmax2d
  from .nms import nms3d
label_desc: Use image node_id 3 for calling the ConvSoftArgmax3d obj's underlying member method code with example usage: obj.forward(x) and returns: conv_soft_argmax3d