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

返回

稀疏張量的稀疏型別。