sugar_hang的博客

在PyTorch中，张量Torch是最基础的运算单位，与Numpy的ndarray相同都表示的是一个多维的矩阵。与ndarray的最大区别就是，PyTorch的Tensor可以在 GPU 上运行，而 numpy 的 ndarray 只能在 CPU 上运行，在GPU上运行大大加快了运算速度。 1、Tensor的数据类型 1.1、创建浮点型Tensor：torch.FloatTensor a = torch.FloatTensor(3, 5) print(a) b = torch.Floa...

转载：Datawhale《60题PyTorch简易入门指南，做技术的弄潮儿！》 PyTorch是一个基于Python的库，提供了一个具有灵活易用的深度学习框架，是近年来最受欢迎的 深度学习框架之一。 一、初识PyTorch 1.1、张量 1.1.1、导入PyTorch包 import torch 1.1.2、创建一个空的5x3张量 x = torch.empty(5,3) print(x) 1.1.3、创建一个随机初始化的5×3张量 x = torch.ran...