![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch
师大先生
在这里记录一些平时学的零碎的小知识哈哈哈~~~
展开
-
PyTorch学习笔记 (一)——入门篇
一、Tensor入门 inplace method: y.add_ (x) equals y = y.add(x) tensor和numpy类型可以快速转换,是共享内存,几乎不消耗时间。但应注意值改变问题。 使用GPU加速 Tensor的.cuda方法可以将其转化为GPU的Tensor 可以发现GPU比CPU快了将近十倍 二、Autograd:自动微分 autograd.Vari...原创 2019-09-28 04:05:35 · 233 阅读 · 0 评论 -
PyTorch学习笔记(二)——Tensor
1、创建Tensor 函数 功能 Tensor(*size) 基础构造函数 ones(*size) 全1 zeros(*size) 全0 eye(*size) 对角线为1,其他为0 arange(begin,end,step) 从begin到end,步长为step linspace(begin,end,steps) 从begin到end,等间隔取steps个值 ...原创 2019-10-01 03:54:48 · 209 阅读 · 0 评论 -
PyTorch学习笔记(三)——autograd
简介:torch.autograd模块能够根据输入和向前传播过程自动构建计算图,并自动求导,执行反向传播。原创 2019-10-04 00:26:12 · 293 阅读 · 0 评论