tensorflow::ops::BatchMatMul

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

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

Выполняется перемножение всех срезов 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[..., :, :])

Аргументы:

• scope: Объект Scope
• x: 2D или более высокое значение с формой [..., r_x, c_x] .
• y: 2D или более высокое значение с формой [..., 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

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

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AdjX(
  bool x
)

Attrs AdjY(
  bool x
)