pytorch 常用线性函数详解

yipeiwu_com7年前Python基础

Pytorch的线性函数主要封装了Blas和Lapack,其用法和接口都与之类似。

常用的线性函数如下:

函数 功能
trace 对角线元素之和(矩阵的迹)
diag 对角线元素
triu/tril 矩阵的上三角/下三角,可指定偏移量
mm/bmm 矩阵乘法,batch的矩阵乘法
t 转置
dot/cross 内积/外积
inverse 求逆矩阵
svd 奇异值分解

注意:矩阵的转置会使存储空间不连续,需调用它的.contiguous方法转为连续。

例如:

import torch as t
b=a.t()
b.is_contiguous()
 
输出:False
 
b=b.contiguous()
b.is_contiguous()
 
输出:True

以上这篇pytorch 常用线性函数详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

Python Web框架之Django框架Model基础详解

本文实例讲述了Python Web框架之Django框架Model基础。分享给大家供大家参考,具体如下: model是关于你的数据的单一的,确定的信息来源。 它包含您正在存储的数据的基本...

浅谈使用Python变量时要避免的3个错误

Python编程中经常遇到一些莫名其妙的错误, 其实这不是语言本身的问题, 而是我们忽略了语言本身的一些特性导致的,今天就来看下使用Python变量时导致的3个不可思议的错误, 以后在编...

符合语言习惯的 Python 优雅编程技巧【推荐】

Python最大的优点之一就是语法简洁,好的代码就像伪代码一样,干净、整洁、一目了然。要写出 Pythonic(优雅的、地道的、整洁的)代码,需要多看多学大牛们写的代码,github 上...

使用python serial 获取所有的串口名称的实例

如下所示: #!/usr/bin/env python # -*- coding: utf-8 -* import serial import serial.tools.list...

python解析发往本机的数据包示例 (解析数据包)

tcp.py 复制代码 代码如下:# -*- coding: cp936 -*-import socketfrom struct import *from time import cti...