tf.raw_ops.BatchMatMul

Умножает фрагменты двух тензоров в партиях.

tf.raw_ops.BatchMatMul(
    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[..., :, :])