程序包 ai.onnxruntime

类 OnnxSparseTensor.CSRCTensor

java.lang.Object

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

ai.onnxruntime.OnnxSparseTensor.CSRCTensor

封闭类

OnnxSparseTensor

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

CSRC 稀疏张量的 Java 端表示。

构造函数概要

构造函数

构造函数	描述
CSRCTensor(java.nio.LongBuffer outerIndices, java.nio.LongBuffer innerIndices, java.nio.Buffer values, long[] denseShape, OnnxJavaType type, long numNonZero)	创建一个适用于构建 ORT 稀疏张量的 CSRC 稀疏张量。

方法概要

限定符和类型	方法	描述
OnnxJavaType	getIndicesType()	稀疏张量的索引类型。
java.nio.LongBuffer	getInnerIndices()	获取内部索引缓冲区。
long[]	getInnerIndicesShape()	获取内部索引的形状。
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

构造函数详细信息

CSRCTensor

public CSRCTensor(java.nio.LongBuffer outerIndices,
                  java.nio.LongBuffer innerIndices,
                  java.nio.Buffer values,
                  long[] denseShape,
                  OnnxJavaType type,
                  long numNonZero)

创建一个适用于构建 ORT 稀疏张量的 CSRC 稀疏张量。

参数

outerIndices - 外部索引。

innerIndices - 内部索引。

values - 数据。

denseShape - 稠密形状。

type - 数据类型。

numNonZero - 非零元素的数量。

方法详细信息

getInnerIndicesShape

public long[] getInnerIndicesShape()

获取内部索引的形状。

返回值

内部索引的形状。

getInnerIndices

public java.nio.LongBuffer getInnerIndices()

获取内部索引缓冲区。

返回值

内部索引缓冲区。

getIndicesType

public OnnxJavaType getIndicesType()

从类复制的描述： OnnxSparseTensor.SparseTensor

稀疏张量的索引类型。

仅支持 OnnxJavaType.INT32 和 OnnxJavaType.INT64。

指定者

getIndicesType 在类中 OnnxSparseTensor.SparseTensor<java.nio.LongBuffer>

返回值

稀疏张量索引类型。

getSparsityType

public OnnxSparseTensor.SparseTensorType getSparsityType()

从类复制的描述： OnnxSparseTensor.SparseTensor

稀疏张量的稀疏度类型。

指定者

getSparsityType 在类中 OnnxSparseTensor.SparseTensor<java.nio.LongBuffer>

返回值

稀疏张量稀疏度类型。