tensoreflusso:: ops:: BatchMatMulV2

Moltiplica le fette di due tensori in batch.

Riepilogo

Moltiplica tutte le sezioni del Tensor x (ogni sezione può essere visualizzata come un elemento di un batch y e dispone i singoli risultati in un singolo tensore di output della stessa dimensione del batch. Ciascuna delle singole fette può facoltativamente essere unita (aggiungere una matrice significa trasporla e coniugarla) prima della moltiplicazione impostando il flag adj_x o adj_y su True , che sono per impostazione predefinita False .

I tensori di input x y 2-D o superiori con forma [..., r_x, c_x] e [..., r_y, c_y] .

Il tensore di output è 2-D o superiore con forma [..., r_o, c_o] , dove:

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

Viene calcolato come:

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

NOTA : BatchMatMulV2 supporta la trasmissione nelle dimensioni batch. Maggiori informazioni sulla trasmissione qui .

Argomenti:

• scope: un oggetto Scope
• x: 2-D o superiore con forma [..., r_x, c_x] .
• y: 2-D o superiore con forma [..., r_y, c_y] .

Attributi facoltativi (vedi Attrs ):

• adj_x: Se True , aggiunge le fette di x . Il valore predefinito è False .
• adj_y: Se True , aggiunge le fette di y . Il valore predefinito è False .

Resi:

• Output : 3-D o superiore con forma [..., r_o, c_o]

Attributi pubblici

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
)