包 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 在類 java.lang.Object 中

      • isScalar

        public boolean isScalar()
        如果形狀表示標量值（即它具有零維度），則返回 true。
        返回
        如果形狀是標量，則為 True。

      • getNumElements

        public long getNumElements()
        返回此張量中的元素數量。

        如果返回值為負數，則此張量資訊指的是輸入或輸出佔位符，該佔位符具有符號維度，並且在未指定符號維度的情況下無法計算元素數量。

        返回
        元素數量。

      • makeCarrier

        public java.lang.Object makeCarrier()
                             throws OrtException
        構造一個具有正確形狀和型別以容納此張量的陣列。

        注意：對於 String 張量，此 carrier 是一個一維陣列，其空間足以容納所有元素，因為這是原生程式碼期望的格式。可以使用 OrtUtil.reshape(String[],long[]) 將其重塑為正確形狀。

        對於 fp16 和 bf16 張量，輸出 carrier 型別為 float，因此此方法生成多維 float 陣列。

        返回
        適當基本型別（或 String）的多維陣列。
        丟擲
        OrtException - 如果形狀無法在 Java 中表示（即如果其中一個索引大於 int）。

      • constructFromJavaArray

        public static TensorInfo constructFromJavaArray​(java.lang.Object obj)
                                         throws OrtException
        從提供的多維 Java 陣列構造一個 TensorInfo，用於分配適當大小的原生記憶體。
        引數
        obj - 要檢查的物件。
        返回
        可用於建立正確大小張量的 TensorInfo。
        丟擲
        OrtException - 如果提供的 Object 不是陣列或型別無效。

      • constructFromBuffer

        public static TensorInfo constructFromBuffer​(java.nio.Buffer buffer,
                                             long[] shape,
                                             OnnxJavaType type)
                                      throws OrtException
        從提供的位元組緩衝區構造一個 TensorInfo。
        引數
        buffer - 要檢查的緩衝區。
        shape - 張量的形狀。
        type - Java 型別。
        返回
        張量的 TensorInfo。
        丟擲
        OrtException - 如果提供的緩衝區與形狀不匹配。