首页 > 技术文章 > pytorch学习笔记

ymec 2020-12-30 19:40 原文

1 pytorch 由facebook在2017年在github上开源

pytorch特性:

简洁性(基于与python编程风格一致),动态计算,visdom,部署不方便。

TensorFlow特性:

接口复杂,静态图,Tensorboard,部署方便(TF serving)

 

动态图:编号程序即可执行

静态图:先搭建好计算图,后运行,允许编译器进行优化,代码编程更复杂,调试不直观、

 

in_place操作

既不允许使用中间变量:

a = a + b

没有产生一个新的变量来保存a+b的值

广播机制:张量参数可以自动扩展为相同大小

广播机制需满足两个条件:

(1)每个张量至少一个维度

(2)迭代维度尺寸时,从尾部的维度开始,维度尺寸
​       或者相等,
      ​ 或者其中一个张量的维度尺寸为 1 ,
      ​ 或者其中一个张量不存在这个维度。

推荐阅读