在Tensorflow / Keras中获取对称矩阵的上三角的最有效方法是什么？

Question

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

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