tensorflow :: ops :: BatchMatMulV2

Multiplica fatias de dois tensores em lotes.

Resumo

Multiplica todas as fatias de Tensor x e y (cada fatia pode ser visto como um elemento de um lote), e providencia os resultados individuais em um único tensor do mesmo tamanho do lote de saída. Cada uma das fatias individuais pode ser opcionalmente anexada (juntar uma matriz significa transpor e conjugá-la) antes da multiplicação, definindo o adj_x ou adj_y como True , que são por padrão False .

Os tensores de entrada x e y são o 2-D ou superior com forma [..., r_x, c_x] e [..., r_y, c_y] .

O tensor de saída é 2-D ou superior com forma [..., r_o, c_o] , onde:

r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y

É calculado como:

output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])

NOTA : BatchMatMulV2 suporta transmissão nas dimensões do lote. Mais sobre transmissão aqui .

Argumentos:

• escopo: um objeto Scope
• x: 2-D ou superior com forma [..., r_x, c_x] .
• y: 2-D ou superior com forma [..., r_y, c_y] .

Atributos opcionais (consulte Attrs ):

• adj_x: se True , junte as fatias de x . O padrão é False .
• adj_y: se True , junte as fatias de y . O padrão é False .

Retorna:

• Output : 3-D ou superior com forma [..., r_o, c_o]

Atributos públicos

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
)