包 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> 中

返回

稀疏张量的稀疏类型。