basic\_ops
==========

.. automodule:: penzai.nn.basic_ops

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree: leaf
      :template: pzclass.rst
   
      CastToDType
      Elementwise
      Softmax
      TanhSoftCap