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

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

        :return: The matrix-matrix product
        :rtype: :class:`nngeometry.object.PMatKFAC`
        """
        prod = dict()
        for layer_id, (a, g) in self.data.items():
            (a_other, g_other) = other.data[layer_id]
            prod[layer_id] = (torch.mm(a, a_other),
                              torch.mm(g, g_other))
        return PMatKFAC(self.generator, data=prod)

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

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

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

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

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