docstr: Validate the dynamic axis generated for each parameters are correct
docstr_tok: ["Validate","the","dynamic","axis","generated","for","each","parameters","are","correct"]
code: def test_infer_dynamic_axis_pytorch(self):
      """
      Validate the dynamic axis generated for each parameters are correct
      """
      from transformers import BertModel
  
      model = BertModel(
          BertConfig.from_pretrained("lysandre/tiny-bert-random")
      )
      tokenizer = BertTokenizerFast.from_pretrained(
          "lysandre/tiny-bert-random"
      )
      self._test_infer_dynamic_axis(model, tokenizer, "pt")
  
imports: import unittest
  from pathlib import Path
  from tempfile import NamedTemporaryFile, TemporaryDirectory
  from transformers import BertConfig, BertTokenizerFast, FeatureExtractionPipeline
  from transformers.convert_graph_to_onnx import convert, ensure_valid_input, generate_identified_filename, infer_shapes, quantize
  from transformers.testing_utils import require_tf, require_tokenizers, require_torch, slow
label_desc: Use image node_id 7 for calling the OnnxExportTestCase obj's underlying member method code with example usage: obj.test_infer_dynamic_axis_pytorch() without return types