包 org.apache.tvm

类 NDArray

java.lang.Object
org.apache.tvm.TVMValue
org.apache.tvm.NDArrayBase
org.apache.tvm.NDArray
public class NDArray extends NDArrayBase
TVM 运行时的轻量级 NDArray 类。

  • 字段摘要

    从类 org.apache.tvm.NDArrayBase 继承的字段

    handle, isView

    从类 org.apache.tvm.TVMValue 继承的字段

    typeCode

  • 方法摘要

    修饰符和类型
    方法
    描述
    byte[]
    asByteArray()
    返回当前数组的复制的扁平 java 数组（行优先）。
    char[]
    asCharArray()
    返回当前数组的复制的扁平 java 数组（行优先）。
    double[]
    asDoubleArray()
    返回当前数组的复制的扁平 java 数组（行优先）。
    float[]
    asFloatArray()
    返回当前数组的复制的扁平 java 数组（行优先）。
    int[]
    asIntArray()
    返回当前数组的复制的扁平 java 数组（行优先）。
    long[]
    asLongArray()
    返回当前数组的复制的扁平 java 数组（行优先）。
    short[]
    asShortArray()
    返回当前数组的复制的扁平 java 数组（行优先）。
    void
    copyFrom(byte[] sourceArray)
    从原生数组复制。
    void
    copyFrom(char[] sourceArray)
    从原生数组复制。
    void
    copyFrom(double[] sourceArray)
    从原生数组复制。
    void
    copyFrom(float[] sourceArray)
    从原生数组复制。
    void
    copyFrom(int[] sourceArray)
    从原生数组复制。
    void
    copyFrom(long[] sourceArray)
    从原生数组复制。
    void
    copyFrom(short[] sourceArray)
    从原生数组复制。
    void
    copyFromRaw(byte[] sourceArray)
    从原始字节数组复制。
    设备
    device()
    获取当前数组的设备。
    static NDArray
    empty(long[] shape)
    在 CPU 上创建给定形状的空 float32 数组。
    static NDArray
    empty(long[] shape, Device dev)
    创建给定形状和设备的空 float32 数组。
    static NDArray
    empty(long[] shape, TVMType dtype)
    在 CPU 上创建给定形状和类型的空数组。
    static NDArray
    empty(long[] shape, TVMType dtype, Device dev)
    创建给定形状、类型和设备的空数组。
    protected void
    finalize()
     
    byte[]
    internal()
    返回当前数组的复制的内部字节数组（行优先）。
    long[]
    shape()
    获取当前 NDArray 的形状。
    long
    size()
    获取当前 NDArray 的总大小。

    从类 org.apache.tvm.NDArrayBase 继承的方法

    asNDArray, copyTo, release

    从类 org.apache.tvm.TVMValue 继承的方法

    asBytes, asDouble, asFunction, asLong, asModule, asString

    从类 java.lang.Object 继承的方法

    clone, equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

  • 方法详情

    • finalize

      protected void finalize() throws Throwable
      覆盖
      finalize 在类 NDArrayBase
      抛出
      Throwable

    • copyFrom

      public void copyFrom(double[] sourceArray)
      从原生数组复制。NDArray 类型必须为 float64
      参数
      sourceArray - 源数据

    • copyFrom

      public void copyFrom(float[] sourceArray)
      从原生数组复制。NDArray 类型必须为 float32
      参数
      sourceArray - 源数据

    • copyFrom

      public void copyFrom(long[] sourceArray)
      从原生数组复制。NDArray 类型必须为 int64
      参数
      sourceArray - 源数据

    • copyFrom

      public void copyFrom(int[] sourceArray)
      从原生数组复制。NDArray 类型必须为 float32
      参数
      sourceArray - 源数据

    • copyFrom

      public void copyFrom(short[] sourceArray)
      从原生数组复制。NDArray 类型必须为 int16
      参数
      sourceArray - 源数据

    • copyFrom

      public void copyFrom(byte[] sourceArray)
      从原生数组复制。NDArray 类型必须为 int8
      参数
      sourceArray - 源数据

    • copyFrom

      public void copyFrom(char[] sourceArray)
      从原生数组复制。NDArray 类型必须为 uint16
      参数
      sourceArray - 源数据

    • copyFromRaw

      public void copyFromRaw(byte[] sourceArray)
      从原始字节数组复制。
      参数
      sourceArray - 源数据

    • shape

      public long[] shape()
      获取当前 NDArray 的形状。
      返回
      表示当前 ndarray 形状的数组

    • size

      public long size()
      获取当前 NDArray 的总大小。
      返回
      当前 NDArray 的大小。

    • asDoubleArray

      public double[] asDoubleArray()
      返回当前数组的复制的扁平 java 数组（行优先）。NDArray 数据类型必须为 float64
      返回
      数组内容的副本。

    • asFloatArray

      public float[] asFloatArray()
      返回当前数组的复制的扁平 java 数组（行优先）。NDArray 数据类型必须为 float32
      返回
      数组内容的副本。

    • asLongArray

      public long[] asLongArray()
      返回当前数组的复制的扁平 java 数组（行优先）。NDArray 数据类型必须为 int64
      返回
      数组内容的副本。

    • asIntArray

      public int[] asIntArray()
      返回当前数组的复制的扁平 java 数组（行优先）。NDArray 数据类型必须为 int32
      返回
      数组内容的副本。

    • asShortArray

      public short[] asShortArray()
      返回当前数组的复制的扁平 java 数组（行优先）。NDArray 数据类型必须为 int16
      返回
      数组内容的副本。

    • asCharArray

      public char[] asCharArray()
      返回当前数组的复制的扁平 java 数组（行优先）。NDArray 数据类型必须为 uint16
      返回
      数组内容的副本。

    • asByteArray

      public byte[] asByteArray()
      返回当前数组的复制的扁平 java 数组（行优先）。NDArray 数据类型必须为 int8
      返回
      数组内容的副本。

    • internal

      public byte[] internal()
      返回当前数组的复制的内部字节数组（行优先）。
      返回
      数组内容的副本。

    • device

      public Device device()
      获取当前数组的设备。
      返回
      设备。

    • empty

      public static NDArray empty(long[] shape, TVMType dtype, Device dev)
      创建给定形状、类型和设备的空数组。
      参数
      shape - 数组的形状。
      dtype - 数组的数据类型。
      dev - 数组的设备。
      返回
      数组 tvm 支持。

    • empty

      public static NDArray empty(long[] shape, TVMType dtype)
      在 CPU 上创建给定形状和类型的空数组。
      参数
      shape - 数组的形状。
      dtype - 数组的数据类型。
      返回
      数组 tvm 支持。

    • empty

      public static NDArray empty(long[] shape)
      在 CPU 上创建给定形状的空 float32 数组。
      参数
      shape - 数组的形状。
      返回
      数组 tvm 支持。

    • empty

      public static NDArray empty(long[] shape, Device dev)
      创建给定形状和设备的空 float32 数组。
      参数
      shape - 数组的形状。
      dev - 数组的设备。
      返回
      数组 tvm 支持。