在TensorFlow中展平包含矢量的2D张量的最佳方法?
python
tensorflow
7
0

将实际上是水平或垂直向量的2D张量扁平化为1D张量的最有效方法是什么?

两者之间在性能方面有区别吗?

tf.reshape(w, [-1])

tf.squeeze(w)

参考资料:
Stack Overflow
收藏
评论
共 1 个回答
高赞 时间 活跃

tf.reshape(w, [-1])tf.squeeze(w)都是“便宜”的,因为它们仅对给定张量的元数据(即形状)进行操作,而不修改数据本身。这两个tf.reshape()内部具有稍微简单的逻辑 ,但是两者的性能应该没有区别。

收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号