pytorch 常用线性函数详解
发布时间:2020-12-17 17:32:41 所属栏目:Python 来源:网络整理
导读:Pytorch的线性函数主要封装了Blas和Lapack,其用法和接口都与之类似。 常用的线性函数如下: 函数 功能 trace 对角线元素之和(矩阵的迹) diag 对角线元素 triu/tril 矩阵的上三角/下三角,可指定偏移量 mm/bmm 矩阵乘法,batch的矩阵乘法 t 转置 dot/cross
Pytorch的线性函数主要封装了Blas和Lapack,其用法和接口都与之类似。 常用的线性函数如下:
注意:矩阵的转置会使存储空间不连续,需调用它的.contiguous方法转为连续。 例如: import torch as t b=a.t() b.is_contiguous() 输出:False b=b.contiguous() b.is_contiguous() 输出:True 以上这篇pytorch 常用线性函数详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持我们。 (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |