pytorch计算矩阵的迹

x=torch.rand(3,3)
print(x)
print(x.trace())#求矩阵的迹（对角线元素之和）；
print(x.diag())#对角线元素之和；
print(x.inverse())#求矩阵的逆；
print(x.triu())#求矩阵的上三角
print(x.tril())#求矩阵的下三角；
print(x.t())#矩阵的转置；
y=torch.rand(3,1)
print(x.mm(y))#矩阵乘法；

pytorch计算矩阵的迹相关推荐

numpy使用diagonal函数和sum函数计算矩阵的迹（trace）、使用T函数对矩阵进行转置（transpose matrix）
numpy使用diagonal函数和sum函数计算矩阵的迹(trace).使用T函数对矩阵进行转置(transpose matrix) 目录
pytorch 计算相似度，相关系数
torch.cosine_similarity 可以对两个向量或者张量计算相似度结果:可能为负数,只能判断方向,相似度精度比较低. 比如1和10000的相似度为1,1和-1的相似度为-1. impo ...
pytorch计算模型参数量
1. 安装 thop 1.1 常规安装 pip install thop 1.2 若上述安装方式错误,可以参考以下方式: pip install thop-i http://pypi.douban.c ...
pytorch计算FLOPs
转自:pytorch计算FLOPs - 简书 1. 引言其实模型的参数量好算,但浮点运算数并不好确定,我们一般也就根据参数量直接估计计算量了.但是像卷积之类的运算,它的参数量比较小,但是运算量非常大 ...
Pytorch计算神经网络的参数量和浮点操作数
参考链接:https://github.com/ShichenLiu/CondenseNet/blob/master/utils.py 计算参数量遍历模型的参数,并提取参数的维度: def get_ ...
pytorch——计算两个等大无序点云中的距离Earth Mover Distance
目前要做两个无序点集之间的相似性计算,在看过Chamfer Distance后,个人觉得CD的计算方式决定了其无法处理两个点集整体分布差异大,但是局部有部分点距离很近的情况,而这种情况在点集中是一定可 ...
pytorch计算余弦相似度
在pytorch中,可以使用torch.cosine_similarity函数对两个向量或者张量计算余弦相似度.先看一下pytorch源码对该函数的定义: class CosineSimilarity ...
pytorch 计算 kl散度 F.kl_div()
先附上官方文档说明:torch.nn.functional - PyTorch 1.13 documentation torch.nn.functional.kl_div(input, target, ...
Pytorch 计算误判率，计算准确率，计算召回率
下面就说说怎么计算准确率以及误判率.召回率等指标 1.计算正确率获取每批次的预判正确个数 train_correct = (pred == batch_y.squeeze(1)).sum() 该语句 ...

pytorch计算矩阵的迹

pytorch计算矩阵的迹相关推荐

最新文章

热门文章