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的矩阵范数或向量范数

pytorch用法笔记--不定期更新相关推荐

Linux个人学习笔记,不定期更新
**如何解决虚拟机正在被使用问题:**将后缀为vmx.lck文件删除 Linux的使用当更改某项配置时,先查看脚本文件有无相关设置.(系统配置脚本文件在/etc,用户配置文件在~) 查看当前目录pw ...
【面试】 C++ 面试题笔记(不定期更新)
题1.关于ASSERT: 题目描述 void func() {char b[2]={0};strcpy(b,"aaaa"); } 以下说法那个正确() A. Debug版崩溃,Re ...
小程序需要php知识吗,小程序和PHP学习笔记 ----- 不定期更新。
学习tp5和小程序过程需要记住的重点记录 1,box-sizing: border-box; 规定两个并排的带边框的框 border-box 为元素设定的宽度和高度决定了元素的边框盒. 就是说,为元素 ...
自动控制原理上课笔记（不定期更新）
文章目录前言自动控制的一般概念经典控制理论和现代控制理论的区别自动控制的基本原理与方式自动控制系统基本控制方式对自动控制系统的基本要求控制系统的数学模型数学模型微分方程小偏差法传 ...
pytorch学习笔记（2）：在MNIST上实现一个CNN
参考文档:https://mp.weixin.qq.com/s/1TtPWYqVkj2Gaa-3QrEG1A 这篇文章是在一个大家经常见到的数据集 MNIST 上实现一个简单的 CNN.我们会基于上一 ...
Pytorch使用笔记
Pytorch使用笔记 nn.Module CNN torch.nn.Conv1d torch.nn.Conv2d torch.nn.ConvTranspose1d RNN LSTM self.reg ...
PyTorch 学习笔记（六）：PyTorch hook 和关于 PyTorch backward 过程的理解 call
您的位置首页 PyTorch 学习笔记系列 PyTorch 学习笔记(六):PyTorch hook 和关于 PyTorch backward 过程的理解发布: 2017年8月4日 7,195阅读 ...
PyTorch学习笔记(二)——回归
PyTorch学习笔记(二)--回归本文主要是用PyTorch来实现一个简单的回归任务. 编辑器:spyder 1.引入相应的包及生成伪数据 import torch import torch.nn ...
Android实用代码（不定期更新）
Android实用代码七段(一) 一抹火焰整理(不定期更新) 前言这里积累了一些不常见确又很实用的代码,每收集7条更新一次,希望能对大家有用. 声明: 欢迎转载,但请保留文章原始出处:) 博客园 ...