В python предположим, что
a = np.array(range(0,12)).reshape(2,2,3)
b = np.array(range(0,6)).reshape(3,2)
c = np.matmul(a,b) // a @ b
У нас есть
a: array([[[ 0, 1, 2],
[ 3, 4, 5]],
[[ 6, 7, 8],
[ 9, 10, 11]]])
b: array([[0, 1],
[2, 3],
[4, 5]])
c: array([[[10, 13],
[28, 40]],
[[46, 67],
[64, 94]]])
Может ли кто-нибудь помочь мне выполнить эквивалентную операцию в java nd4j без цикла for?Я пробовал broadcast.mul
, но оказывается, broadcast.mul
- это поэлементное умножение.Я не нашел ни одной трансляции для mmul.