1、tensorflow中几个关键抽象:(个人初步理解,可能有错误)
节点:一个tensor代表一个节点
计算图:定义的多个tensor之间的运算关系用graph的形式来表达,从而形成一张图
占位符:顾名思义,提前占据计算位置的变量
会话:给占位符传值需要一个人机交互界面,这个东西叫会话
计算路径:提前设定好加减乘除运算规则,各个tensor按照这个路径来进行计算
参考文章:【T-Tensorflow框架学习】Tensorflow “计算图”入门理解_小狐狸-CSDN博客_tensorflow计算图
2、tensorflow中的常用函数:
文章总结:【Tensorflow】tf.argmax函数_zj360202的专栏-CSDN博客_tf.argmax
【转载】TensorFlow 入门必看:Google AI 实习生经验谈(非常适合 tensorflow 初学者理解框架的文章)_tz_zs的博客-CSDN博客
3、slice操作
整体感觉远远不如torch好用,参考如下
tf.slice()函数详解(极详细) - Kobay - 博客园
import tensorflow as tf
sess = tf.Session()
t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]], [[5, 5, 5], [6, 6, 6]]])
a = tf.slice(t, [0,1,0], [3,1,2])
print(f"a is {sess.run(a)}")
b = tf.slice(t, [1,0,0], [2,1,3])
print(f"b is {sess.run(b)}")
结果如下:
a is [[[2 2]]
[[4 4]]
[[6 6]]]
b is [[[3 3 3]]
[[5 5 5]]]