类的用法
org.apache.tvm.NDArray

使用 NDArray 的包
描述
 
  • org.apache.tvmNDArray 的用法

    org.apache.tvm 中返回 NDArray 的方法
    修饰符和类型
    方法
    描述
    static NDArray
    NDArray.empty(long[] shape)
    在 CPU 上根据给定形状创建一个空的 float32 数组。
    static NDArray
    NDArray.empty(long[] shape, Device dev)
    根据给定形状和设备创建一个空的 float32 数组。
    static NDArray
    NDArray.empty(long[] shape, TVMType dtype)
    在 CPU 上根据给定形状和类型创建一个空的数组。
    static NDArray
    NDArray.empty(long[] shape, TVMType dtype, Device dev)
    根据给定形状、类型和设备创建一个空的数组。