Pytorch入门（一）

admin • • 人工智能

Pytorch入门（一）

首先导入pytorch的包

import torch

创建一个5*3的空矩阵

x=torch.empty(5,3)

在这里插入图片描述
创建一个5*3的全零矩阵

x=torch.zeros(5,3)

在这里插入图片描述
创建一个5*3的全一矩阵

a=x.new_ones(5,3)

在这里插入图片描述
根据已有矩阵创建一个形状相同的随机矩阵

a=torch.randn_like(x)

在这里插入图片描述
查看矩阵的形状

x.size()

在这里插入图片描述
矩阵的运算

a+x
torch.add(x,a)
#两种方法的结果一样

在这里插入图片描述

本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。
THE END
分享
二维码
< <上一篇
下一篇>>