class_docstr: Test large corner cases for tensorflow.
class_docstr_tok: ["Test","large","corner","cases","for","tensorflow","."]
docstr: Test that PauliZ with large batched states works as expected.
docstr_tok: ["Test","that","PauliZ","with","large","batched","states","works","as","expected","."]
code: def test_pauliz_large_batched_state_tf(self):
      """Test that PauliZ with large batched states works as expected."""
      import tensorflow as tf
  
      @qml.qnode(qml.device("default.qubit"), interface="tf")
      def circuit(init_state):
          qml.StatePrep(init_state, wires=range(8))
          qml.PauliX(0)
          qml.PauliZ(0)
          return qml.state()
  
      states = np.zeros((3, 256))
      states[:, 0] = 1.0
      results = circuit(tf.Variable(states))
      assert qml.math.shape(results) == (3, 256)
      assert np.array_equal(results[:, 128], [-1.0 + 0.0j] * 3)
  
imports: from functools import reduce
  import pytest
  import numpy
  from scipy.stats import unitary_group
  import pennylane
  from pennylane.operation import _UNSET_BATCH_SIZE, Operation
  from pennylane.devices.qubit.apply_operation import apply_operation, apply_operation_einsum, apply_operation_tensordot
label_desc: Use image node_id 3 for calling the TestLargeTFCornerCases obj's underlying member method code with example usage: obj.test_pauliz_large_batched_state_tf() and returns: qml