tensorflow::ops::BatchMatMulV3

#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[..., :, :])

ПРИМЕЧАНИЕ: BatchMatMulV3 поддерживает широковещательную передачу по измерениям пакета. Подробнее о широковещательной передаче можно узнать здесь.

Аргументы:

• scope: объект Scope
• x: размерность 2 или выше, форма [..., r_x, c_x].
• y: размерность 2 или выше, форма [..., r_y, c_y].
• Tout: если не указано, Tout имеет тот же тип, что и входные данные.

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

• adj_x: если True, выполнить сопряжённое транспонирование срезов x. По умолчанию False.
• adj_y: если True, выполнить сопряжённое транспонирование срезов y. По умолчанию False.

Возвращает:

• Output: размерность 3 или выше, форма [..., r_o, c_o]

Открытые атрибуты

Operation operation

::tensorflow::Output output

Открытые функции

 BatchMatMulV3(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  DataType Tout
)

 BatchMatMulV3(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  DataType Tout,
  const BatchMatMulV3::Attrs & attrs
)

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const 

Открытые статические функции

Attrs AdjX(
  bool x
)

Attrs AdjY(
  bool x
)