code: def test_basic(self) -> None:
      torch.manual_seed(0)
  
      B = 20
      D = 3
      in_features = 10
      dense_arch = DenseArch(
          in_features=in_features, hidden_layer_size=10, embedding_dim=D
      )
  
      dense_arch_input = torch.rand((B, in_features))
      dense_embedded = dense_arch(dense_arch_input)
      self.assertEqual(dense_embedded.size(), (B, D))
  
      # check tracer compatibility
      gm = torch.fx.GraphModule(dense_arch, Tracer().trace(dense_arch))
      script = torch.jit.script(gm)
      script(dense_arch_input)
  
file_path: torchrec/torchrec/models/tests/test_deepfm.py
imports: import unittest
  import torch
  from torch.testing import FileCheck
  from torchrec.fx import symbolic_trace, Tracer
  from torchrec.models.deepfm import DenseArch, FMInteractionArch, SimpleDeepFMNN
  from torchrec.modules.embedding_configs import EmbeddingBagConfig
  from torchrec.modules.embedding_modules import EmbeddingBagCollection
  from torchrec.sparse.jagged_tensor import KeyedJaggedTensor, KeyedTensor
label_desc: Use image node_id 1 for calling the DenseArchTest obj's underlying member method code with example usage: obj.test_basic() without return types