code: def test_normal():
      loc = torch.as_tensor([0.3, 0.5])
      scale = torch.as_tensor([0.1, 0.9])
      dist = torch.distributions.Normal(loc, scale)
      mode = mode_of_distribution(dist)
      torch_assert_allclose(mode, loc)
  
file_path: pfrl/tests/utils_tests/test_mode_of_distribution.py
imports: import math
  import torch
  from pfrl.testing import torch_assert_allclose
  from pfrl.utils.mode_of_distribution import mode_of_distribution
label_desc: Use image node_id 3 for calling a global function with example usage: test_normal() without return types