美文网首页
TF 重要函数记录(1)

TF 重要函数记录(1)

作者: 飘曳的舟 | 来源:发表于2019-01-02 21:31 被阅读0次

最近在研究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进行转置操作

相关文章

网友评论

      本文标题:TF 重要函数记录(1)

      本文链接:https://www.haomeiwen.com/subject/xxbtrqtx.html