public class Module
extends java.lang.Object
| 修飾詞和類型 | 方法和說明 |
|---|---|
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)
使用指定的引數執行此模組的指定方法。
|
public static Module load(java.lang.String modelPath, java.util.Map<java.lang.String,java.lang.String> extraFiles, Device device)
public static Module load(java.lang.String modelPath)
modelPath - 包含序列化 TorchScript 模組的檔案路徑。Module 物件。public IValue forward(IValue... inputs)
inputs - TorchScript 模組「forward」方法的引數。public IValue runMethod(java.lang.String methodName, IValue... inputs)
methodName - 要執行的 TorchScript 方法名稱。inputs - 將傳遞至 TorchScript 方法的引數。public void destroy()
destroy 可以更快釋放記憶體。請參閱 HybridData.resetNative()。