【材料整理】 -- torch中常用知识点。

文章目录

写在前面

本文主要总结自己毕设过程中，用到torch库中一些知识点，之后会不断补充，欢迎大家收藏！

import torch

1、torch.where函数 跟双目运算符一样；
torch.where(condition,a,b) condition为真，执行a；否则，执行b。

2、torch.numpy()将Tensor变量转换为ndarray变量，其中t是一个Tensor变量，可以是标量，也可以是向量，转换后dtype与Tensor的dtype一致。

3、.cpu() 将数据的处理设备从其他设备（如.cuda()拿到cpu上），不会改变变量类型，转换后仍然是Tensor变量。

4、.cuda()函数将Tensor加载到gpu中运行

5、squeeze()和unsqueeze()函数
torch.squeeze() 这个函数主要对数据的维度进行压缩，去掉维数为1的的维度，比如是一行或者一列这种，一个一行三列（1,3）的数去掉第一个维数为一的维度之后就变成（3）行。
torch.unsqueeze()这个函数主要是对数据维度进行扩充。给指定位置加上维数为一的维度，比如原本有个三行的数据（3），在0的位置加了一维就变成一行三列（1,3）。

6、torch.log()函数
7、permute函数
8、torch.ones_like()函数

9、把numpy数组转换成tensor形式
10、torch.cuda函数（与GPU相关）