tfep.nn.transformers.sos.SOSPolynomialTransformerFunc
=====================================================

.. currentmodule:: tfep.nn.transformers.sos

.. autoclass:: SOSPolynomialTransformerFunc
   :members:
   :show-inheritance:
   :inherited-members: Module,LightningModule

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~SOSPolynomialTransformerFunc.__init__
      ~SOSPolynomialTransformerFunc.apply
      ~SOSPolynomialTransformerFunc.backward
      ~SOSPolynomialTransformerFunc.forward
      ~SOSPolynomialTransformerFunc.get_sos_poly_coefficients
      ~SOSPolynomialTransformerFunc.jvp
      ~SOSPolynomialTransformerFunc.mark_dirty
      ~SOSPolynomialTransformerFunc.mark_non_differentiable
      ~SOSPolynomialTransformerFunc.mark_shared_storage
      ~SOSPolynomialTransformerFunc.maybe_clear_saved_tensors
      ~SOSPolynomialTransformerFunc.name
      ~SOSPolynomialTransformerFunc.register_hook
      ~SOSPolynomialTransformerFunc.register_prehook
      ~SOSPolynomialTransformerFunc.save_for_backward
      ~SOSPolynomialTransformerFunc.save_for_forward
      ~SOSPolynomialTransformerFunc.set_materialize_grads
      ~SOSPolynomialTransformerFunc.setup_context
      ~SOSPolynomialTransformerFunc.vjp
      ~SOSPolynomialTransformerFunc.vmap
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~SOSPolynomialTransformerFunc.dirty_tensors
      ~SOSPolynomialTransformerFunc.generate_vmap_rule
      ~SOSPolynomialTransformerFunc.materialize_grads
      ~SOSPolynomialTransformerFunc.metadata
      ~SOSPolynomialTransformerFunc.needs_input_grad
      ~SOSPolynomialTransformerFunc.next_functions
      ~SOSPolynomialTransformerFunc.non_differentiable
      ~SOSPolynomialTransformerFunc.requires_grad
      ~SOSPolynomialTransformerFunc.saved_for_forward
      ~SOSPolynomialTransformerFunc.saved_tensors
      ~SOSPolynomialTransformerFunc.saved_variables
      ~SOSPolynomialTransformerFunc.to_save