Как разделить 2d-матрицу по указанной позиции ⇐ Python

[Anonymous]

У меня есть карта объектов, которую нужно разделить на другую константу 2d, карта объектов имеет форму [H, W, ...] , константа 2d имеет форму [H, W] , однако, когда я использую np.divide, он говорит, что эти два тензора не могут быть преобразованы в общую форму:

Код: Выделить всё

import numpy as np

a = np.random.rand(4, 3 ,3)
b = np.random.rand(4, 3)

c = np.divide(a, b, where=b > 0)

как говорят правила трансляции numpy, b будет транслироваться в форме [1, 4, 3] вместо [4, 3, 1] , поэтому я не могу выполнить эту операцию с помощью boradcast.
Я хотел бы разделить мою карту объектов с помощью этого постоянного тензора, где константа не равна 0, чтобы избежать ошибки деления на ноль, что мне следует сделать?

Подробнее здесь: https://stackoverflow.com/questions/792 ... d-position