тензорный поток:: опс:: ПакетныйМатМулВ2

Пакетно умножает срезы двух тензоров.

Краткое содержание

Умножает все срезы 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[..., :, :])

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

Аргументы:

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

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

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

Возврат:

• Output : 3D или выше с формой [..., 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
)