《深度学习之PyTorch实战计算机视觉》学习笔记（2）

最新推荐文章于 2023-07-12 07:12:38 发布

Blessing_l

最新推荐文章于 2023-07-12 07:12:38 发布

阅读量714

点赞数

分类专栏：计算机视觉文章标签： python pytorch

本文链接：https://blog.csdn.net/weixin_40017911/article/details/89014336

版权

这篇博客主要介绍了使用PyTorch 1.0进行计算机视觉的实战，基于python3.7和CUDA 10.0进行开发。

摘要由CSDN通过智能技术生成

这部分是关于pytorch的基本知识。代码基于python3.7， pytorch 1.0，cuda 10.0 .

import torch
a = torch.FloatTensor(2,3)    # pytorch 定义数据类型的方式，可以输入一个维度值或者列表
b = torch.FloatTensor([2,3,4,5])
# print(a)
# print(b)
c = torch.IntTensor(2,3)
d = torch.IntTensor([2,3,4,5])
print(c)
print(d)

tensor([[          0,           0,   183490976],
        [        652,           0, -2147483648]], dtype=torch.int32)
tensor([2, 3, 4, 5], dtype=torch.int32)