tensorflow::ops::BatchMatMulV2

#include <math_ops.h>

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

Краткое описание

Умножает все фрагменты тензора Tensorx и 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 поддерживает вещание по размерам партии. Подробнее о вещании здесь.

Аргументы:

• scope: Объект Scope
• x: 2-мерный или выше с формой [..., r_x, c_x].
• y: 2-мерный или выше с формой [..., r_y, c_y].

Необязательные атрибуты (см. Attrs):

• adj_x: Если True, сопрягает фрагменты x. По умолчанию False.
• adj_y: Если True, сопрягает фрагменты y. По умолчанию False.

Возвращает:

• Output: 3-мерный или выше с формой [..., r_o, c_o]

Общедоступные атрибуты

Operation operation

::tensorflow::Output output

Общедоступные функции

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y
)

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  const BatchMatMulV2::Attrs & attrs
)

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const 

Общедоступные статические функции

Attrs AdjX(
  bool x
)

Attrs AdjY(
  bool x
)