随笔分类 - tensorflow
pytorch和tensorflow
摘要:举个简单的例子,当我们要实现一个这样的计算图时: 用TensorFlow是这样的: 而用pytorch是这样的: 当然,里面都包含了建立前向计算图,传入变量数据,求梯度等操作,但是显而易见,pytorch的代码更为凝练。TensorFlow我也用得比较少,我本文重点说说pytorch的一些学习心得,
阅读全文
浙公网安备 33010602011771号