pytorch用法笔记--不定期更新

torch.matmul(a,a.T)

矩阵乘以它的转置矩阵，如3*4,4*3，相乘后结果是一个3*3的对称矩阵

torch.diag(a, 2)

取以对角线为起点的元素，对角线的位置为0，2代表从对角线开始往上的第2条对角线，也可以是负值，也就是对角线往下

torch.cat((a, b), dim=0)

拼接两个矩阵，按照dim=0的维度，比如3*4,3*5，拼接后就是3*9，dim=0也就是横向拼接，dim=1也就是纵向拼接

np.fill_diagonal(a,0)/torch.fill_diagonal(a,0)

一个np,一个pytorch的用法，都是填充对角线元素为0

torch.norm(input, p='fro', dim=None, keepdim=False, out=None, dtype=None)

返回所给tensor的矩阵范数或向量范数