包 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

返回

稀疏张量稀疏性类型。