tfep.nn.masked.MaskedLinearFunc
===============================

.. currentmodule:: tfep.nn.masked

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

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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