code: def log(input: DTensor):
      return DTensor(
          input.shardings.map_shard(
              torch.log, dtype_promote_to=torch.float32
          )
      )
  
imports: import torch
  from ._tensor import DTensor, implements
label_desc: Use image node_id 2 for calling a global function with example usage: log(input) and returns: DTensor