aliran tensor:: operasi:: BatchMatMulV2

#include <math_ops.h>

Mengalikan irisan dua tensor secara berkelompok.

Ringkasan

Mengalikan semua irisan Tensor x dan y (setiap irisan dapat dilihat sebagai elemen suatu kumpulan), dan menyusun hasil individual dalam satu tensor keluaran dengan ukuran kumpulan yang sama. Masing-masing irisan individual dapat digabungkan secara opsional (menyatukan matriks berarti mengubah posisi dan mengkonjugasikannya) sebelum perkalian dengan menyetel tanda adj_x atau adj_y ke True , yang secara default adalah False .

Tensor input x dan y adalah 2-D atau lebih tinggi dengan bentuk [..., r_x, c_x] dan [..., r_y, c_y] .

Tensor keluarannya adalah 2-D atau lebih tinggi dengan bentuk [..., r_o, c_o] , dengan:

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

Itu dihitung sebagai:

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

CATATAN : BatchMatMulV2 mendukung penyiaran dalam dimensi batch. Lebih lanjut tentang penyiaran di sini .

Argumen:

  • ruang lingkup: Objek Lingkup
  • x: 2-D atau lebih tinggi dengan bentuk [..., r_x, c_x] .
  • y: 2-D atau lebih tinggi dengan bentuk [..., r_y, c_y] .

Atribut opsional (lihat Attrs ):

  • adj_x: Jika True , satukan irisan x . Defaultnya adalah False .
  • adj_y: Jika True , satukan irisan y . Defaultnya adalah False .

Pengembalian:

  • Output : 3-D atau lebih tinggi dengan bentuk [..., r_o, c_o]

Konstruktor dan Destruktor

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)

Atribut publik

operation
output

Fungsi publik

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

Fungsi statis publik

AdjX (bool x)
AdjY (bool x)

Struktur

tensorflow:: ops:: BatchMatMulV2:: Attrs

Penyetel atribut opsional untuk BatchMatMulV2 .

Atribut publik

operasi

Operation operation

keluaran

::tensorflow::Output output

Fungsi publik

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
)

simpul

::tensorflow::Node * node() const 

operator::tensorflow::Masukan

 operator::tensorflow::Input() const 

operator::tensorflow::Keluaran

 operator::tensorflow::Output() const 

Fungsi statis publik

AdjX

Attrs AdjX(
  bool x
)

AdjY

Attrs AdjY(
  bool x
)