1. 本际云推荐 - 专业推荐VPS、服务器,IDC点评首页
  2. 云主机运维
  3. VPS运维

探讨tensorflow与pytorch的互相转变

简介

我是本际云服务器推荐网的小编小本本。在以下文章中,将简单介绍tensorflow与pytorch的互相转换,对学习培训或工作有很强的参考和使用价值,希望能够与大家一起学习。

探讨tensorflow与pytorch的互相转变

tensorflow与pytorch的转换

在以下编码实例中,将简单介绍tensorflow转换成pytorch。由于自己只了解pytorch,对tensorflow不得而知,存在许多不正确可能,请轻喷~

转换详解

1.自变量形式参数:在tensorflow中,变量定义和重置是分离的,而在pytorch中,定义和变量初始化是拼在一起的。

2.建立自变量并重置:在tensorflow中引入tf.Variable创建自变量并重置,而pytorch使用torch.tensor创建自变量并重置。

3.句子实行:在tensorflow中,所有关于图自变量的取值和测算都需要根据tf.Session的run去进行。而在pytorch中,不一定需要调用run,可以直接取值后立即测算。

4.tensor:在pytorch中,需要将numpy数组转换为tensor进行运算,最后再将tensor转换回numpy数组。而在tensorflow中不需要这种操作。

5.其他函数:tensorflow中包含一些pytorch没有的函数,但是都可以在其他库中找到相似的函数。

以上就是关于tensorflow转换成pytorch的简单介绍,希望对大家有所帮助!

原创文章,作者:小编小本本,如若转载,请注明出处:https://www.benjiyun.com/yunzhujiyunwei/vps-yunwei/6646.html