code: def test_soft_round_conditional_mean():
      for offset in range(-5, 5):
          x = torch.linspace(offset + 0.001, offset + 0.999, 100)
  
          torch.testing.assert_close(
              torch.round(x),
              soft_round_conditional_mean(x, alpha=5000.0),
              atol=0.001,
              rtol=0.001,
          )
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_soft_round_conditional_mean() without return types