code: def get_scalars(
      input_tensor: torch.Tensor, t_step: int, with_bs: bool = True
  ):
      """Adds the time step to the current state tensor.
  
      Args:
          input_tensor (torch.Tensor): Current state tensor.
          t_step (int): Current time step.
          with_bs (bool, optional): Whether the batch size is present in the
          input tensor.
      """
      # scalars containing the iteration time
      if with_bs:
          bs = input_tensor.shape[0]
          scalars = torch.zeros((bs, 1))
          scalars[:, 0] = t_step
      else:
          scalars = torch.tensor(t_step).unsqueeze(-1).float()
      return scalars
  
imports: from typing import Tuple
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: get_scalars(input_tensor, t_step, with_bs) and returns: scalars