TF 重要函数记录(1)

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

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容