tf.raw_ops.BatchMatMulV3

Умножает куски двух тензоров в пакетах.

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

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

Входные тензоры x и y являются двумерными или многомерными с формами [..., r_x, c_x] и [..., r_y, c_y].

Выходной тензор является двумерным или многомерным с формой [..., 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[..., :, :])

Примечание: BatchMatMulV3 поддерживает вещание в размерностях пакета. Подробнее о вещании здесь.