.. _readout: Readout blocks ============== .. module:: mlip.models.readout .. autoclass:: MultiHeadReadoutBlock .. automethod:: __call__ .. autoclass:: ReadoutBlock .. automethod:: __call__ .. autofunction:: select_head