程序包 ai.onnxruntime

类 OnnxSparseTensor.BlockSparseTensor

java.lang.Object

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

ai.onnxruntime.OnnxSparseTensor.BlockSparseTensor

封闭类

OnnxSparseTensor

public static final class OnnxSparseTensor.BlockSparseTensor
extends OnnxSparseTensor.SparseTensor<java.nio.IntBuffer>

块稀疏张量的 Java 端表示。

构造函数概要

构造函数

构造函数	描述
BlockSparseTensor(java.nio.IntBuffer indices, long[] indicesShape, java.nio.Buffer values, long[] valuesShape, long[] denseShape, OnnxJavaType type, long numNonZero)	构造一个块稀疏张量。

方法概要

限定符和类型	方法	描述
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

构造函数详细信息

BlockSparseTensor

public BlockSparseTensor(java.nio.IntBuffer indices,
                         long[] indicesShape,
                         java.nio.Buffer values,
                         long[] valuesShape,
                         long[] denseShape,
                         OnnxJavaType type,
                         long numNonZero)

构造一个块稀疏张量。

参数

indices - 索引。

indicesShape - 索引的形状。

values - 数据。

valuesShape - 数据的形状。

denseShape - 稠密形状。

type - 数据类型。

numNonZero - 非零元素的数量。

方法详细信息

getIndicesType

public OnnxJavaType getIndicesType()

从类复制的描述： OnnxSparseTensor.SparseTensor

稀疏张量的索引类型。

仅支持 OnnxJavaType.INT32 和 OnnxJavaType.INT64。

指定者

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

返回值

稀疏张量索引类型。

getSparsityType

public OnnxSparseTensor.SparseTensorType getSparsityType()

从类复制的描述： OnnxSparseTensor.SparseTensor

稀疏张量的稀疏度类型。

指定者

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

返回值

稀疏张量稀疏度类型。