Pomnóż macierz „a” przez macierz „b”.
Dane wejściowe muszą być macierzami dwuwymiarowymi, a wewnętrzny wymiar „a” musi odpowiadać zewnętrznemu wymiarowi „b”. Zarówno „a”, jak i „b” muszą być „Tensorem”, a nie „SparseTensorem”. Ta operacja jest zoptymalizowana pod kątem przypadku, gdy co najmniej jedno z „a” lub „b” jest rzadkie, w tym sensie, że mają one dużą część wartości zerowych. Próg rentowności przy zastosowaniu tego w porównaniu z mnożeniem gęstej macierzy na jednej platformie wynosił 30% wartości zerowych w macierzy rzadkiej.
Obliczenie gradientu w tej operacji będzie uwzględniać rzadkość gradientu wejściowego tylko wtedy, gdy gradient ten pochodzi z Relu.
Klasy zagnieżdżone
klasa | Opcje SparseMatMul | Opcjonalne atrybuty dla SparseMatMul |
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
statyczne SparseMatMul.Options | aIsSparse (logiczna aIsSparse) |
Dane wyjściowe <TFloat32> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
statyczne SparseMatMul.Options | bIsSparse (logiczna bIsSparse) |
statyczny SparseMatMul | |
Dane wyjściowe <TFloat32> | produkt () |
statyczne SparseMatMul.Options | transpozycjaA (logiczna transpozycjaA) |
statyczne SparseMatMul.Options | transpozycja B (logiczna transpozycja B) |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
publiczne wyjście < TFloat32 > asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static SparseMatMul create (zakres zakresu , Operand <? rozszerza TNumber > a, Operand <? rozszerza TNumber > b, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację SparseMatMul.
Parametry
zakres | aktualny zakres |
---|---|
opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja SparseMatMul