code: def test_load_and_save(self):
      linear = Linear(10, 2)()
      sigmoid = Sigmoid()(linear)
      softmax = SoftMax().set_name("output")(sigmoid)
      model_original = Model([linear], [softmax])
      input = np.random.random((4, 10))
  
      temp = tempfile.mkdtemp()
  
      model_original.save_tensorflow(
          [("input", [4, 10])], temp + "/model.pb"
      )
  
      model_loaded = Model.load_tensorflow(
          temp + "/model.pb", ["input"], ["output"]
      )
      model_loaded_without_backwardgraph = Model.load_tensorflow(
          temp + "/model.pb",
          ["input"],
          ["output"],
          generated_backward=False,
      )
      expected_output = model_original.forward(input)
      output = model_loaded.forward(input)
      output_without_backwardgraph = (
          model_loaded_without_backwardgraph.forward(input)
      )
      assert_allclose(output, expected_output, atol=1e-6, rtol=0)
      assert_allclose(
          output_without_backwardgraph,
          expected_output,
          atol=1e-6,
          rtol=0,
      )
  
file_path: BigDL/python/dllib/test/bigdl/test_tensorflow.py
imports: from bigdl.dllib.nn.layer import *
  from bigdl.dllib.optim.optimizer import *
  from bigdl.dllib.utils.common import *
  import numpy
  import unittest
  import shutil
  import tempfile
  from numpy.testing import assert_allclose
label_desc: Use image node_id 1 for calling the TestTensorflow obj's underlying member method code with example usage: obj.test_load_and_save() without return types