最近在研究python实现的任务式对话管理,想用TF来实现,python版本的TF相比pytorch 在debug难度,代码实现复杂度上明显是要高的,但是目前TF在工业界的使用明显还是要广于pytorch,因此决定还是用TF来进行实现。
下面先记录几个用到的高频函数
-
tf.layers.dense 一个dense层计算,内嵌了W矩阵,只需要输入和output dims
-
tf.matmul 矩阵乘法,十分强大,对于带batch_size的两个tensor相乘,可以保持batchsize 不变。
-
tf.tile 对tensor进行重复填充的操作,multiplier可以是一个和维度对应的list,表示对应维度重复的倍数。
-
tf.expand_dims 升维操作
-
tf.transpose 对tensor进行转置操作
网友评论