code: def aligned_array(size, dtype, align=64):
      
  
      n = size * dtype.itemsize
      empty = np.empty(n + (align - 1), dtype=np.uint8)
      data_align = empty.ctypes.data % align
      offset = 0 if data_align == 0 else (align - data_align)
      output = empty[offset:offset + n].view(dtype)
  
      assert len(output) == size, len(output)
      assert output.ctypes.data % align == 0, output.ctypes.data
      return output
docs: Returns an array of a given size that is 64-byte aligned.
  
      The returned array can be efficiently copied into GPU memory by TensorFlow.