码迷,mamicode.com
首页 > 其他好文 > 详细

Tensorflow模型训练时维度处理技巧

时间:2021-02-05 10:33:17      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:tip   multi   sign   维度   模型   name   个数   assign   val   

Tensorflow模型训练时维度处理技巧

修改模型一周,调BUG半个月,特此总结:

  1. 在某一个维度为1时,用tf.reshape
  2. tf.matmul(要求两个元素维度正好相反要求两个元素维度正好相反
  3. tf.multiply ( 张量1 ,张量2)要求两个元素维度一致
  4. assign_sub():重新赋值,但需要先用 tf.Variable 定义变量 w
  5. 要求输出的数据格式保持一致且支持广播作用的函数:
  6. tf.muliply(x, y, name=None)  
  7. tf.add(x, y, name=None)
  8. tf.concat(values, axis, name=‘concat‘)多种网络joint训练的时候逆向操作tf.split(value, num_or_size_splits, axis=0, num=None, name=‘split‘)
  9. tf.stack(values, axis, name=‘stack‘)增加指定维度
  10. tf.tile()把数组沿各个方向复制不变的维度设置为1
  11. 反之用tf.squeeze压缩维度
  12. tf.nn.embedding_lookup(ids = [1,3,5]):找出embeddings中第1,3,5行,组成一个tensor返回,其维度是有几个数则输出几个值
  13. tf.split(value,num_or_size_splits,axis=0)
  14. value张量沿着axis维度,按照num_or_size_splits个数切分

 

Tensorflow模型训练时维度处理技巧

标签:tip   multi   sign   维度   模型   name   个数   assign   val   

原文地址:https://www.cnblogs.com/sweet-dreamer/p/14372084.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!