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

tensorflow :: ops :: SparseMatMul

#include <math_ops.h>

"A" matrisini "b" matrisiyle çarpın .

Özet

Girdiler iki boyutlu matrisler olmalı ve "a" nın iç boyutu "b" nin dış boyutuyla eşleşmelidir. Hem "a" hem de "b" Tensor olmalı, SparseTensor değil. Bu işlem, "a" veya "b" den en az birinin, büyük oranda sıfır değerine sahip olmaları bakımından seyrek olduğu durum için optimize edilmiştir. Bunu bir platformda yoğun matris çarpımına karşı kullanmak için başabaş, seyrek matriste% 30 sıfır değerleriydi.

Bu işlemin gradyan hesaplaması, yalnızca gradyan bir Relu'dan geldiğinde giriş gradyanındaki seyreklikten yararlanacaktır.

Argümanlar:

  • kapsam: Bir Scope nesnesi

İadeler:

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

SparseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b)
SparseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b, const SparseMatMul::Attrs & attrs)

Genel özellikler

operation
product

Kamusal işlevler

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

Genel statik işlevler

AIsSparse (bool x)
BIsSparse (bool x)
TransposeA (bool x)
TransposeB (bool x)

Yapılar

tensorflow :: ops :: SparseMatMul :: Attrs

SparseMatMul için isteğe bağlı öznitelik belirleyiciler.

Genel özellikler

operasyon

Operation operation

ürün

::tensorflow::Output product

Kamusal işlevler

SparseMatMul

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

SparseMatMul

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const SparseMatMul::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

AIsSparse

Attrs AIsSparse(
  bool x
)

BIsSparse

Attrs BIsSparse(
  bool x
)

TransposeA

Attrs TransposeA(
  bool x
)

TransposeB

Attrs TransposeB(
  bool x
)