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 · 240 阅读 · 0 评论 -
PyTorch学习笔记(二)——Tensor
1、创建Tensor函数功能Tensor(*size)基础构造函数ones(*size)全1zeros(*size)全0eye(*size)对角线为1,其他为0arange(begin,end,step)从begin到end,步长为steplinspace(begin,end,steps)从begin到end,等间隔取steps个值...原创 2019-10-01 03:54:48 · 213 阅读 · 0 评论 -
PyTorch学习笔记(三)——autograd
简介:torch.autograd模块能够根据输入和向前传播过程自动构建计算图,并自动求导,执行反向传播。原创 2019-10-04 00:26:12 · 303 阅读 · 0 评论