Поиск элемента в одном массиве, соответствующего максимальному значению в другом

Как я могу выровнять массив 4d z и массив 4d QCLOUD и затем узнать значение z, когда происходит QCLOUD max?

print(z.shape) print(qcloud.shape) out: (6, 100, 128, 128) (6, 99, 128, 128) 

Python - лучший язык программирования в мире.