包 ai.onnxruntime

類 OnnxSparseTensor.COOTensor

java.lang.Object

ai.onnxruntime.OnnxSparseTensor.SparseTensor<java.nio.LongBuffer>

ai.onnxruntime.OnnxSparseTensor.COOTensor

封閉類

OnnxSparseTensor

public static final class OnnxSparseTensor.COOTensor
extends OnnxSparseTensor.SparseTensor<java.nio.LongBuffer>

COO 稀疏張量的 Java 端表示。

構造器摘要

構造器

構造器	描述
COOTensor(java.nio.LongBuffer indices, long[] indicesShape, java.nio.Buffer values, long[] denseShape, OnnxJavaType type, long numNonZero)	建立一個適合構建 ORT 稀疏張量的 COO 稀疏張量。

方法摘要

修飾符和型別	方法	描述
OnnxJavaType	getIndicesType()	稀疏張量的索引型別。
OnnxSparseTensor.SparseTensorType	getSparsityType()	稀疏張量的稀疏性型別。

繼承自類 ai.onnxruntime.OnnxSparseTensor.SparseTensor 的方法

getDenseShape, getIndices, getIndicesShape, getNumNonZeroElements, getType, getValues, getValuesShape

繼承自類 java.lang.Object 的方法

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

構造器詳細資訊

COOTensor

public COOTensor(java.nio.LongBuffer indices,
                 long[] indicesShape,
                 java.nio.Buffer values,
                 long[] denseShape,
                 OnnxJavaType type,
                 long numNonZero)

建立一個適合構建 ORT 稀疏張量的 COO 稀疏張量。

引數

indices - 索引。應為 1d 向量或 2d 向量。

indicesShape - 索引的形狀。

values - 資料。

denseShape - 密集形狀。

type - 資料型別。

numNonZero - 非零元素的數量。

方法詳細資訊

getIndicesType

public OnnxJavaType getIndicesType()

描述複製自類: OnnxSparseTensor.SparseTensor

稀疏張量的索引型別。

僅支援 OnnxJavaType.INT32 和 OnnxJavaType.INT64。

指定者

類 OnnxSparseTensor.SparseTensor<java.nio.LongBuffer> 中的 getIndicesType

返回

稀疏張量索引型別。

getSparsityType

public OnnxSparseTensor.SparseTensorType getSparsityType()

描述複製自類: OnnxSparseTensor.SparseTensor

稀疏張量的稀疏性型別。

指定者

類 OnnxSparseTensor.SparseTensor<java.nio.LongBuffer> 中的 getSparsityType

返回

稀疏張量稀疏性型別。