code: def testShape(self):
      batch_size = 1
      num_writes = 2
      memory_size = 5
      word_size = 3
  
      mem = tf.random.uniform([batch_size, memory_size, word_size])
      write_address = tf.random.uniform(
          [batch_size, num_writes, memory_size]
      )
      reset_weights = tf.random.uniform(
          [batch_size, num_writes, word_size]
      )
      writer = write.erase(mem, write_address, reset_weights)
      self.assertTrue(
          writer.shape.as_list(), [batch_size, memory_size, word_size]
      )
  
file_path: sonnet/sonnet/src/nets/dnc/write_test.py
imports: import numpy
  from sonnet.src import test_utils
  from sonnet.src.nets.dnc import write
  import tensorflow
label_desc: Use image node_id 1 for calling the EraseTest obj's underlying member method code with example usage: obj.testShape() without return types