nngeometry/object/pspace.py
def mm(self, other):
        """
        Matrix-matrix product where `other` is another 
        instance of PMatBlockDiag

        :param other: Other FIM matrix
        :type other: :class:`nngeometry.object.PMatBlockDiag`

        :return: The matrix-matrix product
        :rtype: :class:`nngeometry.object.PMatBlockDiag`
        """
        prod = dict()
        for layer_id, block in self.data.items():
            block_other = other.data[layer_id]
            prod[layer_id] = torch.mm(block, block_other)
        return PMatBlockDiag(self.generator,
                             data=prod)

        Matrix-matrix product where `other` is another 
        instance of PMatBlockDiag

        :param other: Other FIM matrix
        :type other: :class:`nngeometry.object.PMatBlockDiag`

        :return: The matrix-matrix product
        :rtype: :class:`nngeometry.object.PMatBlockDiag`
        
Matrix-matrix product where `other` is another
instance of PMatBlockDiag
Matrix - matrix product where ` other ` is another instance of PMatBlockDiag
def mm(self, other):
        prod = dict()
        for layer_id, block in self.data.items():
            block_other = other.data[layer_id]
            prod[layer_id] = torch.mm(block, block_other)
        return PMatBlockDiag(self.generator,
                             data=prod)
def mm ( self , other ) : prod = dict ( ) for layer_id , block in self . data . items ( ) : block_other = other . data [ layer_id ] prod [ layer_id ] = torch . mm ( block , block_other ) return PMatBlockDiag ( self . generator , data = prod )
Matrix-matrix product where `other` is another
instance of PMatBlockDiag
Matrix - matrix product where ` other ` is another instance of PMatBlockDiag
"""
        Matrix-matrix product where `other` is another 
        instance of PMatBlockDiag

        :param other: Other FIM matrix
        :type other: :class:`nngeometry.object.PMatBlockDiag`

        :return: The matrix-matrix product
        :rtype: :class:`nngeometry.object.PMatBlockDiag`
        """
The matrix-matrix product
The matrix - matrix product
:class:`nngeometry.object.PMatBlockDiag`
self
other
Other FIM matrix
Other FIM matrix