org.pytorch

类 PyTorchAndroid

java.lang.Object

org.pytorch.PyTorchAndroid

public final class PyTorchAndroid
extends java.lang.Object

构造函数摘要

构造函数

构造函数和描述
PyTorchAndroid()

方法摘要

修饰符和类型	方法和描述
static Module	loadModuleFromAsset(android.content.res.AssetManager assetManager, java.lang.String assetName)
static Module	loadModuleFromAsset(android.content.res.AssetManager assetManager, java.lang.String assetName, Device device) 注意：这并不是加载生产模块的推荐方法，因为预打包的资产会增加 apk 大小等。
static void	setNumThreads(int numThreads) 全局设置原生端使用的线程数。

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

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

构造函数详情

PyTorchAndroid

public PyTorchAndroid()

方法详情

loadModuleFromAsset

public static Module loadModuleFromAsset(android.content.res.AssetManager assetManager,
                                         java.lang.String assetName,
                                         Device device)

注意：这并不是加载生产模块的推荐方法，因为预打包的资产会增加 apk 大小等。对于生产使用，请考虑使用从磁盘文件加载 Module.load(String)。

此方法旨在用于测试和演示。

loadModuleFromAsset

public static Module loadModuleFromAsset(android.content.res.AssetManager assetManager,
                                         java.lang.String assetName)

setNumThreads

public static void setNumThreads(int numThreads)

全局设置原生端使用的线程数。注意：这将产生全局影响，所有模块都使用具有指定线程数的单个线程池。

参数

numThreads - 线程数，必须为正数。