org.pytorch

类 Module

java.lang.Object

org.pytorch.Module

public class Module
extends java.lang.Object

torch::jit::Module 的 Java 包装器。

方法摘要

修饰符和类型	方法和描述
void	destroy() 显式销毁本机 torch::jit::Module。
IValue	forward(IValue... inputs) 使用指定参数运行此模块的“forward”方法。
static Module	load(java.lang.String modelPath) 从磁盘上指定路径加载序列化 TorchScript 模块，以便在 CPU 上运行。
static Module	load(java.lang.String modelPath, java.util.Map<java.lang.String,java.lang.String> extraFiles, Device device) 从磁盘上指定路径加载序列化 TorchScript 模块，以便在指定设备上运行。
IValue	runMethod(java.lang.String methodName, IValue... inputs) 使用指定参数运行此模块的指定方法。

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

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

方法详情

load

public static Module load(java.lang.String modelPath,
                          java.util.Map<java.lang.String,java.lang.String> extraFiles,
                          Device device)

从磁盘上指定路径加载序列化 TorchScript 模块，以便在指定设备上运行。

参数

modelPath - 包含序列化 TorchScript 模块的文件路径。

extraFiles - 键为额外文件名称，值为其内容的映射。

device - 用于运行指定模块的 Device。

返回

拥有 torch::jit::Module 的新的 Module 对象。

load

public static Module load(java.lang.String modelPath)

从磁盘上指定路径加载序列化 TorchScript 模块，以便在 CPU 上运行。

参数

modelPath - 包含序列化 TorchScript 模块的文件路径。

返回

拥有 torch::jit::Module 的新的 Module 对象。

forward

public IValue forward(IValue... inputs)

使用指定参数运行此模块的“forward”方法。

参数

inputs - TorchScript 模块“forward”方法的参数。

返回

“forward”方法的返回值。

runMethod

public IValue runMethod(java.lang.String methodName,
                        IValue... inputs)

使用指定参数运行此模块的指定方法。

参数

methodName - 要运行的 TorchScript 方法的名称。

inputs - 将传递给 TorchScript 方法的参数。

返回

方法的返回值。

destroy

public void destroy()

显式销毁本机 torch::jit::Module。调用此方法不是必需的，因为当此对象被垃圾回收时，本机对象将被销毁。但是，垃圾回收的时机无法保证，因此主动调用 destroy 可以更快地释放内存。请参阅 HybridData.resetNative()。