程序包 ai.onnxruntime

类 TensorInfo

  • 所有已实现的接口
    ValueInfo
    public class TensorInfo
extends java.lang.Object
implements ValueInfo
    描述一个 OnnxTensor，包括其大小、形状和元素类型。

    • 字段详细信息

      • MAX_DIMENSIONS

        public static final int MAX_DIMENSIONS
        Java 接口方法支持的最大维度数。
        另请参阅
        常量字段值

      • type

        public final OnnxJavaType type
        该张量的 Java 类型。

    • 方法详细信息

      • getShape

        public long[] getShape()
        获取张量形状的副本。
        返回值
        张量形状的副本。

      • getDimensionNames

        public java.lang.String[] getDimensionNames()
        获取张量命名维度的副本。
        返回值
        张量命名维度的副本。

      • toString

        public java.lang.String toString()
        重写
        toString in class java.lang.Object

      • isScalar

        public boolean isScalar()
        如果形状表示标量值（即它具有零个维度），则返回 true。
        返回值
        如果形状是标量，则为 True。

      • getNumElements

        public long getNumElements()
        返回该张量中的元素数量。

        如果返回值为负数，则表示该张量信息引用的是具有符号维度的输入或输出占位符，且如果不指定符号维度，则无法计算元素计数。

        返回值
        元素数量。

      • makeCarrier

        public java.lang.Object makeCarrier()
                             throws OrtException
        构造一个具有正确形状和类型以容纳此张量的数组。

        注意：对于字符串张量，此载体是一个单维数组，具有容纳所有元素的足够空间，因为这是原生代码的预期格式。可以使用 OrtUtil.reshape(String[],long[]) 将其重塑为正确的形状。

        对于 fp16 和 bf16 张量，输出载体类型为 float，因此该方法生成多维 float 数组。

        返回值
        适当原始类型（或 String）的多维数组。
        抛出
        OrtException - 如果该形状在 Java 中不可表示（即如果其索引之一大于 int）。

      • constructFromJavaArray

        public static TensorInfo constructFromJavaArray​(java.lang.Object obj)
                                         throws OrtException
        根据提供的多维 Java 数组构造一个 TensorInfo，用于分配适当数量的原生内存。
        参数
        obj - 要检查的对象。
        返回值
        一个可用于创建正确大小 Tensor 的 TensorInfo。
        抛出
        OrtException - 如果提供的对象不是数组，或者类型无效。

      • constructFromBuffer

        public static TensorInfo constructFromBuffer​(java.nio.Buffer buffer,
                                             long[] shape,
                                             OnnxJavaType type)
                                      throws OrtException
        根据提供的字节缓冲区构造一个 TensorInfo。
        参数
        buffer - 要检查的缓冲区。
        shape - 张量的形状。
        type - Java 类型。
        返回值
        一个用于张量的 TensorInfo。
        抛出
        OrtException - 如果提供的缓冲区与形状不匹配。