pytorch学习笔记

一、Tensor运算操作

Tensor是PyTorch中最基本的数据结构，可以将其视为多维数组或者矩阵。PyTorch tensor和NumPy array非常相似，但是tensor还可以在GPU上运算，而NumPy array则只能在CPU上运算。下面，将会整理一些Tensor的操作

首先需要导入pytorch库

import torch

然后就是创建一个Tensor，创建Tensor大概有以下四种方法：

x=torch.empty(5,3)#创建一个未初始化的5*3的矩阵，注意使用时一定要进行初始化后再使用
print(x)

x=torch.rand(5,3)#创建一个随机初始化的矩阵
print(x)

x=torch.zero(5,3)
y=torch.one(5,3)#创建一个全为0/1的矩阵
print(x)
print(y)

x=torch.([1.0,3.0])#直接输入创建一个矩阵
print(x)

查看Tensor的维度

x=torch.rand(5,3)
print(x.shape)

Tensor的操作

x=torch.one(5,3)
y=torch.one(5,3)
z=torch.rand(3,5)
#普通的加减乘除
q=x+1,y=y-1
print(q)
print(y)
q=q/2
print(q)
q=(x+5)*q
print(q)#这里的乘是指叉乘，两个*代表指数
#如果用叉乘则@
x=x@z #x和z做叉乘，维度需要满足叉乘规则
x.add_(1)#如果调用函数做运算后面加_则会直接改变x的值，否则就只做运算