【问题标题】:What is the most efficient way to get the upper triangle of a symmetric matrix in Tensorflow/Keras?在 Tensorflow/Keras 中获取对称矩阵的上三角形的最有效方法是什么?
【发布时间】:2019-05-16 19:32:21
【问题描述】:

我有一个输出对称数组的中间模型。然后将这些数组用作另一个模型的输入。我只想展平阵列并丢弃较低的三角形,因为它们是对称的。有没有最好/最有效的方法来做到这一点?

编辑:我希望三角形提取的处理与任何其他 Keras 层类似,以便第一个模型的输出可以直接输入到第二个模型并进行端到端训练。

【问题讨论】:

标签: python tensorflow keras


【解决方案1】:

Tensorflow 和 Keras 使用 Numpy 来处理数据。考虑使用 Numpy 函数 triutriu_indices

【讨论】:

  • 谢谢。我无法直接使用 NumPy 函数,但 triu 源代码引导我使用 Lambda 层高效实现。
  • @JeremiahJohnson 你能分享你的解决方案吗?谢谢
猜你喜欢
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 2013-07-05
  • 2011-12-18
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 2016-02-15
相关资源
最近更新 更多