tf.raw_ops.BatchMatMulV2

Умножает слайсы двух тензоров в группах.

tf.raw_ops.BatchMatMulV2(
    x, y, adj_x=False, adj_y=False, name=None
)

Умножает все слайсы Tensor x и y (каждый слайс можно рассматривать как элемент группы), и упорядочивает отдельные результаты в одном тензоре вывода с таким же размером группы. Каждый из отдельных слайсов может быть дополнительно сопряжён (сопряжение матрицы означает её транспонирование и комплексное сопряжение) перед умножением, установив флаг adj_x или adj_y в значение True, которые по умолчанию равны False.

Вводные тензоры x и y имеют размерность 2 или выше с формой [..., r_x, c_x] и [..., r_y, c_y].

Тензор вывода имеет размерность 2 или выше с формой [..., r_o, c_o], где:

r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y

Вычисление:

output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])

Примечание: BatchMatMulV2 поддерживает вещание по размерностям группы. Подробнее о вещании здесь.