class_docstr: Test that amplitude embedding gates are combined into a single.
docstr: Test that merging preserves the batch dimension
docstr_tok: ["Test","that","merging","preserves","the","batch","dimension"]
code: def test_broadcasting(self):
      """Test that merging preserves the batch dimension"""
      dev = qml.device("default.qubit", wires=3)
  
      @qml.transforms.merge_amplitude_embedding
      @qml.qnode(dev)
      def qnode():
          qml.AmplitudeEmbedding([[1, 0], [0, 1]], wires=0)
          qml.AmplitudeEmbedding([1, 0], wires=1)
          qml.AmplitudeEmbedding([[0, 1], [1, 0]], wires=2)
          return qml.state()
  
      res = qnode()
      assert qnode.tape.batch_size == 2
  
      # |001> and |100>
      expected = np.array(
          [[0, 1, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 1, 0, 0, 0]]
      )
      assert np.allclose(res, expected)
  
file_path: pennylane/tests/transforms/test_optimization/test_merge_amplitude_embedding.py
imports: import pytest
  from pennylane import numpy
  import pennylane
  from pennylane.transforms.optimization import merge_amplitude_embedding
  from pennylane._device import DeviceError
label_desc: Use image node_id 5 for calling the TestMergeAmplitudeEmbedding obj's underlying member method code with example usage: obj.test_broadcasting() and returns: qml