ML Topluluk Günü 9 Kasım! TensorFlow, JAX güncellemeler için bize katılın ve daha fazla bilgi edinin

tensorflow :: ops :: BatchMatMulV2

#include <math_ops.h>

Gruplar halinde iki tensörün dilimlerini çarpar.

Özet

Tensor x ve y tüm dilimlerini çarpar (her dilim bir partinin bir öğesi olarak görülebilir) ve tek tek sonuçları aynı parti boyutuna sahip tek bir çıktı tensörü olarak düzenler. Çarpma adj_x önce, adj_x veya adj_y bayrağının varsayılan olarak False olan True ayarlanmasıyla, tek tek dilimlerin her biri isteğe bağlı olarak birleştirilebilir (bir matrisin yerini değiştirmek ve birleştirmek anlamına gelir).

Giriş tensörleri x ve y , [..., r_x, c_x] ve [..., r_y, c_y] şeklinde 2-D veya daha yüksektir.

Çıkış tensörü, [..., r_o, c_o] şeklinde 2-D veya daha yüksek, burada:

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

Şu şekilde hesaplanır:

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

NOT : BatchMatMulV2 , toplu iş boyutlarında yayını destekler. Burada yayın yapma hakkında daha fazla bilgi.

Argümanlar:

  • kapsam: Bir Scope nesnesi
  • x: 2-D veya üzeri [..., r_x, c_x] .
  • y: 2-D veya daha yüksek [..., r_y, c_y] .

İsteğe bağlı özellikler (bkz. Attrs ):

  • adj_x: True , x dilimlerini birleştirin. False öntanımlıdır.
  • adj_y: True , y dilimlerini birleştirin. False varsayılanları.

İadeler:

  • Output : 3 boyutlu veya daha yüksek, şekilli [..., r_o, c_o]

Yapıcılar ve Yıkıcılar

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)

Genel özellikler

operation
output

Kamusal işlevler

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

Genel statik işlevler

AdjX (bool x)
AdjY (bool x)

Yapılar

tensorflow :: ops :: BatchMatMulV2 :: Attrs

BatchMatMulV2 için isteğe bağlı öznitelik ayarlayıcılar.

Genel özellikler

operasyon

Operation operation

çıktı

::tensorflow::Output output

Kamusal işlevler

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
)

düğüm

::tensorflow::Node * node() const 

operator :: tensorflow :: Girdi

 operator::tensorflow::Input() const 

operator :: tensorflow :: Çıktı

 operator::tensorflow::Output() const 

Genel statik işlevler

AdjX

Attrs AdjX(
  bool x
)

AdjY

Attrs AdjY(
  bool x
)