Paket ai.onnxruntime

Klasse OnnxSparseTensor

java.lang.Object

ai.onnxruntime.OnnxTensorLike

ai.onnxruntime.OnnxSparseTensor

Alle implementierten Schnittstellen

OnnxValue, java.lang.AutoCloseable

public final class OnnxSparseTensor
extends OnnxTensorLike

Ein Java-Objekt, das einen OnnxSparseTensor umschließt.

Sparse Tensoren unterstützen eine Vielzahl von Formaten, und die Methode getValue() gibt eine andere statische innere Klasse zurück, die jeden Typ repräsentiert.

Zusammenfassung der verschachtelten Klassen

Verschachtelte Klassen

Modifikator und Typ	Klasse	Beschreibung
static Klasse	OnnxSparseTensor.BlockSparseTensor	Die Java-Darstellung eines Block-Sparse-Tensors.
static Klasse	OnnxSparseTensor.COOTensor	Die Java-seitige Darstellung eines COO-Sparse-Tensors.
static Klasse	OnnxSparseTensor.CSRCTensor	Die Java-seitige Darstellung eines CSRC-Sparse-Tensors.
static Klasse	OnnxSparseTensor.SparseTensor<T extends java.nio.Buffer>	Abstrakte Basisklasse für Java Sparse Tensors
static Klasse	OnnxSparseTensor.SparseTensorType	Der Typ des Sparse-Tensors.

Verschachtelte Klassen/Schnittstellen geerbt von der Schnittstelle ai.onnxruntime.OnnxValue

OnnxValue.OnnxValueType

Zusammenfassung der Felder

Von der Klasse geerbte Felder ai.onnxruntime.OnnxTensorLike

allocatorHandle, closed, info, nativeHandle

Zusammenfassung der Methoden

Modifikator und Typ	Methode	Beschreibung
void	close()	Schließt den OnnxValue und gibt seinen nativen Speicher frei.
statisch <T extends java.nio.Buffer> OnnxSparseTensor	createSparseTensor(OrtEnvironment env, OnnxSparseTensor.SparseTensor<T> tensor)	Erstellt einen Sparse Tensor in ORT aus der Java-Seiten-Darstellung.
java.nio.Buffer	getIndicesBuffer()	Ruft eine Kopie der Indizes ab.
long[]	getIndicesShape()	Ruft die Form der (äußeren) Indizes ab.
java.nio.LongBuffer	getInnerIndicesBuffer()	Ruft eine Kopie der inneren Indizes in einem CSRC Sparse Tensor ab.
long[]	getInnerIndicesShape()	Ruft die Form der inneren Indizes in einem CSRC Sparse Tensor ab.
OnnxSparseTensor.SparseTensorType	getSparseTensorType()	Gibt den Typ dieses OnnxSparseTensor zurück.
OnnxValue.OnnxValueType	getType()	Ruft den Typ dieses OnnxValue ab.
OnnxSparseTensor.SparseTensor<? extends java.nio.Buffer>	getValue()	Gibt den Wert als Java-Objekt zurück, indem er aus dem nativen Heap kopiert wird.
java.nio.Buffer	getValuesBuffer()	Ruft einen Puffer der Daten ab.
long[]	getValuesShape()	Ruft die Form der Werte ab.

Von der Klasse geerbte Methoden ai.onnxruntime.OnnxTensorLike

checkClosed, getInfo, isClosed

Methoden geerbt von Klasse java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Detail der Methoden

createSparseTensor

public static <T extends java.nio.Buffer> OnnxSparseTensor createSparseTensor(OrtEnvironment env,
                                                                              OnnxSparseTensor.SparseTensor<T> tensor)
                                                                       throws OrtException

Erstellt einen Sparse Tensor in ORT aus der Java-Seiten-Darstellung.

Typparameter

T - Der Puffertyp.

Parameter

env - Die OrtEnvironment.

tensor - Die Java-Seiten-Darstellung.

Rückgabe

Der Sparse Tensor in ORT.

Wirft

OrtException - Wenn der Tensor nicht erstellt werden konnte oder ungültig war.

getType

public OnnxValue.OnnxValueType getType()

Beschreibung kopiert von der Schnittstelle: OnnxValue

Ruft den Typ dieses OnnxValue ab.

Rückgabe

Der Werttyp.

getValue

public OnnxSparseTensor.SparseTensor<? extends java.nio.Buffer> getValue()
                                                                  throws OrtException

Beschreibung von der Schnittstelle kopiert: OnnxValue

Gibt den Wert als Java-Objekt zurück, indem er aus dem nativen Heap kopiert wird. Dieser Vorgang kann für hochdimensionale Tensoren ziemlich langsam sein, wo Sie stattdessen OnnxTensor.getByteBuffer() usw. verwenden sollten.

Von den Unterklassen überschrieben, falls ein spezifischerer Typ verfügbar ist.

Rückgabe

Der Wert.

Wirft

OrtException - Wenn beim Lesen des Werts ein Fehler aufgetreten ist.

close

public void close()

Beschreibung von der Schnittstelle kopiert: OnnxValue

Schließt den OnnxValue und gibt seinen nativen Speicher frei.

getSparseTensorType

public OnnxSparseTensor.SparseTensorType getSparseTensorType()

Gibt den Typ dieses OnnxSparseTensor zurück.

Rückgabe

Der Sparsity-Typ.

getIndicesBuffer

public java.nio.Buffer getIndicesBuffer()

Ruft eine Kopie der Indizes ab.

Dies sind die äußeren Indizes, wenn es sich um einen CSRC Sparse Tensor handelt.

Es ist ein LongBuffer, wenn COO oder CSRC, und IntBuffer, wenn Block Sparse.

Rückgabe

Die Indizes.

getInnerIndicesBuffer

public java.nio.LongBuffer getInnerIndicesBuffer()

Ruft eine Kopie der inneren Indizes in einem CSRC Sparse Tensor ab.

Löst IllegalStateException aus, wenn sie auf einen anderen Sparse Tensor Typ aufgerufen wird.

Rückgabe

Die inneren Indizes.

getValuesBuffer

public java.nio.Buffer getValuesBuffer()

Ruft einen Puffer der Daten ab.

Wie bei OnnxTensor werden fp16-Werte in fp32 hochskaliert und als FloatBuffer zurückgegeben.

Rückgabe

Der Datenpuffer.

getIndicesShape

public long[] getIndicesShape()

Ruft die Form der (äußeren) Indizes ab.

Rückgabe

Die Indizes-Form.

getInnerIndicesShape

public long[] getInnerIndicesShape()

Ruft die Form der inneren Indizes in einem CSRC Sparse Tensor ab.

Rückgabe

Die Indizes-Form.

getValuesShape

public long[] getValuesShape()

Ruft die Form der Werte ab.

Rückgabe

Die Werte-Form.