当前位置:   article > 正文

(1)pytorch之张量创建_生成一个torch

生成一个torch

一、张量(Tensor)是什么?

1 定义:一个多维数组,是标量(0维张量),向量(1维张量),矩阵(2维张量)的高维拓展。
2 属性:data,dtype(张量的数据类型) ,shape(张量的形状,如64,3,224,224),device(张量所在设备,Gpu/cpu加速关键) 前四个与数据相关
requires_grad , grad ,grad_fn ,is_leaf 与数据求导相关

二、张量的创建

1. 直接创建

直接创建:torch.tensor() :从data来创建tensor,data可以为list/numpy device可以为cuda/cpu
在这里插入图片描述
从numpy创建tensor:torch.from_numpy(ndarry).注:创建的tensor与原ndarray共享内存,改一个,另一个也会改

2.依据数值创建

x=t.empty(5,3)#构造一个53的矩阵,不初始化
y=t.rand(5,6)#构造一个5
6的随机矩阵
z=t.zeros(5,6,dtype=t.float64)#构造一个56的全0矩阵
z=t.tensor(列表)#将这个列表变为张量形式

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Cpp五条/article/detail/716999
推荐阅读
相关标签
  

闽ICP备14008679号