pytorch\_pfn\_extras.nn.modules.ensure\_shape
=============================================

.. automodule:: pytorch_pfn_extras.nn.modules.ensure_shape

   
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
      :toctree: .

      
      pytorch_pfn_extras.nn.modules.ensure_shape.ensure
   
   

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree: .

      
      pytorch_pfn_extras.nn.modules.ensure_shape.Ensure