首页 > 解决方案 > 在 Python 中查找 4D 数组中的对角线索引

问题描述

我有这个形状的 4D 数组,(100,90,90,1)我想找到对角线元素的索引。

我试过numpy.diag_indices_from但得到这个错误:“ValueError:输入的所有维度必须是相等的长度”。

有人可以告诉我实现此目的的其他选择吗?

提前致谢!

标签: arraysindicesdiagonal4d

解决方案


推荐阅读