Pytorch用法说明

1. pytorch基本用法

1. torch.tensor

  • torch.tensor是默认的tensor类型(torch.Floattensor)的简称。

2. torch.Tensor与torch.tensor的区别

  • 相同点: 两者都是用于生成新的张量tensor,其中tensor是一种数据类型。包含单一数据类型元素的多维矩阵。
  • 不同点: torch.Tensor是python类,默认张量类型为torch.FloatTensor的别名。
  • 例如,torch.Tensor([1,2])会调用Tensor类的构造函数 init,生成floattensor类型的张量。 torch.tensor仅仅是python的函数,主要是将一个data封装成一个tensor。
  • 其函数原型是: torch.tensor(data,dtype=None,device=None,requires_grad=False) 其中data可以是 list || tuple || array || scalar等类型。
  • torch.tensor可以从data中的数据部分做拷贝copy(而不是直接引用),根据原始数据类型生成相应的torch.LongTensor、torch.FloatTensor、torch.DoubleTensor。

3. transforms.ToTensor与transforms.ToPILImage

  • torchvision.transforms中定义了一系列数据转换形式。有PILImage、numpy,Tensor之间的转换,还能对数据进行处理。
  • ToTensor:将PILImage或者numpy的ndarry转化成tensor。取值范围[0,1.0]
    +
    1. 对于PILImage转化的tensor,其数据类型为torch.FloatTensor
    2. 把一个取值范围是[0,255]的PIL.Image转换成tensor
    3. img=Image.open()</li> <li>out_img=transforms.ToTensor(img)
      +
    4. 对于ndarray的数据类型没有限制,但转化成tensor的数据类型是由ndarry的数据类型决定的。
    5. shape为(H,W,C)的numpy.ndarray转换成形状为(C,H,W)、取值范围是[0,1.0]的tensor
  • ToPILImage:将Pnumpy的ndarry或者tensor转换成PILImage。PILImage取值范围为[0,255]
    +
    1. ndarry的数据类型要求是dtype=uint8,取值范围[0,255],shape为(H,W,C)
      +
    2. tensor的shape为CHW,只能是FLoatTensor类型。
  • 相关链接:Pytorch之浅入torchvision.transforms.ToTensor与ToPILImage_啧啧啧biubiu的博客-CSDN博客_transforms.topilimagePytorch之浅入torchvision.transforms.ToTensor与ToPILImage Pytorch之浅入torchvision.transforms.ToTensor与ToPILImage_啧啧啧biubiu的博客-CSDN博客_transforms.topilimage

2. Pytorch错误类型

1. 模型的 &#x5206;&#x7C7B;&#x6570;&#x4E0D;&#x5339;&#x914D;

RuntimeError: CUDA error:CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)

2. 标签越界

RuntimeError: CUDA error: device-side assert triggered

Solution:

  1. 首先检查自己的标签是否从0开始;
  2. 将模型、损失函数、张量全部转到CPU上面运行或者debug,此时会报错具体的位置。因为cpu比cuda更能够定位自己的错误所在。在debug模式下,我发现的确是因为某个索引越界导致程序异常(但是cuda始终告诉我是标签越界)

3. pytorch中梯度求导只能针对FolatTensor类型吗?

Yes 关于这个问题如果有人了解,麻烦请告知一声,主要小编也不太清楚,只不过是根据所查资料得出的结论,若有错误,请加以指正。

若有错误,请加以指正,感谢。

Original: https://blog.csdn.net/weixin_44857576/article/details/123758372
Author: weixin_44857576
Title: Pytorch用法说明

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/759562/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球