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()。