przepływ tensorowy:: ops:: BatchMatMulV2

#include <math_ops.h>

Mnoży wycinki dwóch tensorów w partiach.

Streszczenie

Mnoży wszystkie wycinki Tensor x i y (każdy wycinek można postrzegać jako element wsadu) i porządkuje poszczególne wyniki w pojedynczy tensor wyjściowy o tym samym rozmiarze wsadu. Każdy z poszczególnych wycinków można opcjonalnie połączyć (łączenie macierzy oznacza jej transpozycję i koniugację) przed mnożeniem, ustawiając flagę adj_x lub adj_y na True , która domyślnie ma wartość False .

Tensory wejściowe x i y są dwuwymiarowe lub wyższe i mają kształt [..., r_x, c_x] i [..., r_y, c_y] .

Tensor wyjściowy jest dwuwymiarowy lub wyższy i ma kształt [..., r_o, c_o] , gdzie:

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

Jest obliczany jako:

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

UWAGA : BatchMatMulV2 obsługuje rozgłaszanie w wymiarach wsadowych. Więcej o transmisji tutaj .

Argumenty:

  • zakres: Obiekt Scope
  • x: 2-D lub wyższy z kształtem [..., r_x, c_x] .
  • y: 2-D lub wyższy z kształtem [..., r_y, c_y] .

Opcjonalne atrybuty (patrz Attrs ):

  • adj_x: Jeśli True , połącz plasterki x . Domyślnie ustawiona jest False .
  • adj_y: Jeśli True , połącz plasterki y . Domyślnie ustawiona jest wartość False .

Zwroty:

  • Output : 3-D lub wyższe z kształtem [..., r_o, c_o]

Konstruktory i destruktory

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)

Atrybuty publiczne

operation
output

Funkcje publiczne

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

Publiczne funkcje statyczne

AdjX (bool x)
AdjY (bool x)

Struktury

tensorflow:: ops:: BatchMatMulV2:: Attrs

Opcjonalne ustawiacze atrybutów dla BatchMatMulV2 .

Atrybuty publiczne

operacja

Operation operation

wyjście

::tensorflow::Output output

Funkcje publiczne

BatchMatMulV2

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

BatchMatMulV2

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

węzeł

::tensorflow::Node * node() const 

operator::tensorflow::Wejście

 operator::tensorflow::Input() const 

operator::tensorflow::Wyjście

 operator::tensorflow::Output() const 

Publiczne funkcje statyczne

Przym.X

Attrs AdjX(
  bool x
)

Przym.Y

Attrs AdjY(
  bool x
)