code: def _simple_gather_all_tensors(
      result: Tensor, group: Any, world_size: int
  ) -> List[Tensor]:
      gathered_result = [
          torch.zeros_like(result) for _ in range(world_size)
      ]
      torch.distributed.all_gather(gathered_result, result, group)
      return gathered_result
  
imports: import contextlib
  import logging
  import os
  import time
  from contextlib import nullcontext
  from pathlib import Path
  from typing import TYPE_CHECKING, Any, Iterable, Iterator, List, Optional, Sized, Union
  import torch
  import torch.nn.functional
  from lightning_utilities.core.imports import package_available
  from torch import Tensor
  from torch.utils.data import Dataset, DistributedSampler, Sampler
  from typing_extensions import override
  from lightning.fabric.utilities.cloud_io import _is_local_file_protocol
  from lightning.fabric.utilities.data import _num_cpus_available
  from lightning.fabric.utilities.rank_zero import rank_zero_info
  from lightning.fabric.utilities.types import _PATH, ReduceOp
label_desc: Use image node_id 3 for calling a global function with example usage: _simple_gather_all_tensors(result, group, world_size) and returns: gathered_result