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

#include <math_ops.h>

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

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

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

Конструкторы и деструкторы

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)

Публичные атрибуты

operation
output

Общественные функции

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Публичные статические функции

AdjX (bool x)
AdjY (bool x)

Структуры

tensorflow:: ops:: BatchMatMulV2:: Attrs

Дополнительные установщики атрибутов для BatchMatMulV2 .

Публичные атрибуты

операция

Operation operation

выход

::tensorflow::Output output

Общественные функции

ПакетныйМатМулВ2

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

ПакетныйМатМулВ2

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

узел

::tensorflow::Node * node() const 

оператор::tensorflow::Input

 operator::tensorflow::Input() const 

оператор::tensorflow::Выход

 operator::tensorflow::Output() const 

Публичные статические функции

AdjX

Attrs AdjX(
  bool x
)

AdjY

Attrs AdjY(
  bool x
)