docstr: Check that all the NeuralStackCell tensor shapes are correct.
      
docstr_tok: ["Check","that","all","the","NeuralStackCell","tensor","shapes","are","correct","."]
code: def test_cell_shapes(self):
      """Check that all the NeuralStackCell tensor shapes are correct."""
      batch_size = 5
      embedding_size = 4
      memory_size = 12
      num_units = 8
  
      deque = neural_stack.NeuralDequeCell(
          num_units, memory_size, embedding_size
      )
      deque.build(None)
  
      self.assertEqual(
          [1, 1, memory_size, memory_size], deque.get_read_mask(0).shape
      )
      self.assertEqual(
          [1, 1, memory_size, memory_size], deque.get_read_mask(1).shape
      )
  
      deque_input = tf.zeros(
          [batch_size, 1, embedding_size], dtype=tf.float32
      )
      zero_state = deque.zero_state(batch_size, tf.float32)
      (outputs, (deque_next_state)) = deque.call(
          deque_input, zero_state
      )
  
      # Make sure that deque output shapes match deque input shapes
      self.assertEqual(outputs.shape, deque_input.shape)
  
      assert_cell_shapes(self, deque_next_state, zero_state)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import mock
  import numpy
  from tensor2tensor.layers import modalities
  from tensor2tensor.models.research import neural_stack
  from tensor2tensor.utils import contrib
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling the NeuralDequeCellTest obj's underlying member method code with example usage: obj.test_cell_shapes() without return types