code: def _get_flatten_input(inputs: List[torch.Tensor]) -> torch.Tensor:
      return torch.cat(
          [input.flatten(1) for input in inputs],
          dim=1,
      )
  
file_path: torchrec/torchrec/modules/deepfm.py
imports: from typing import List
  import torch
  from torch import nn
  from torch.fx import wrap
label_desc: Use image node_id 1 for calling a global function with example usage: _get_flatten_input(inputs) and returns: torch