ai.djl.pytorch.engine

Class PtSymbolBlock

java.lang.Object

ai.djl.pytorch.jni.NativeResource

ai.djl.pytorch.engine.PtSymbolBlock

All Implemented Interfaces:

ai.djl.nn.Block, ai.djl.nn.SymbolBlock, java.lang.AutoCloseable

public class PtSymbolBlock
extends NativeResource
implements ai.djl.nn.SymbolBlock

PtSymbolBlock is the PyTorch implementation of SymbolBlock.

You can create a PtSymbolBlock using Model.load(java.nio.file.Path, String).

Field Summary

Fields inherited from class ai.djl.pytorch.jni.NativeResource

handle

Constructor Summary

Constructors

Constructor and Description
PtSymbolBlock(PtNDManager manager, Pointer handle) Constructs a PtSymbolBlock.

Method Summary

Modifier and Type	Method and Description
void	cast(ai.djl.ndarray.types.DataType dataType)
void	clear()
void	close()
ai.djl.util.PairList<java.lang.String,ai.djl.ndarray.types.Shape>	describeInput()
ai.djl.ndarray.NDList	forward(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, boolean training, ai.djl.util.PairList<java.lang.String,java.lang.Object> params)
ai.djl.nn.BlockList	getChildren()
java.util.List<ai.djl.nn.Parameter>	getDirectParameters()
ai.djl.ndarray.types.Shape[]	getOutputShapes(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.types.Shape[] inputShapes)
ai.djl.nn.ParameterList	getParameters()
ai.djl.ndarray.types.Shape	getParameterShape(java.lang.String name, ai.djl.ndarray.types.Shape[] inputShapes)
ai.djl.ndarray.types.Shape[]	initialize(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.types.DataType dataType, ai.djl.ndarray.types.Shape... inputShapes)
boolean	isInitialized()
void	loadParameters(ai.djl.ndarray.NDManager manager, java.io.DataInputStream is)
void	removeLastBlock()
void	saveParameters(java.io.DataOutputStream os)
void	setInitializer(ai.djl.training.initializer.Initializer initializer)
void	setInitializer(ai.djl.training.initializer.Initializer initializer, java.lang.String paramName)

Methods inherited from class ai.djl.pytorch.jni.NativeResource

finalize, getHandle, getUid, isReleased

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.nn.Block

forward, validateLayout

Constructor Detail

PtSymbolBlock

public PtSymbolBlock(PtNDManager manager,
                     Pointer handle)

Constructs a PtSymbolBlock.

You can create a PtSymbolBlock using Model.load(java.nio.file.Path, String).

Parameters:

manager - the manager to use for the block

handle - the module handle

Method Detail

close

public void close()

Specified by:

close in interface java.lang.AutoCloseable

Overrides:

close in class NativeResource

removeLastBlock

public void removeLastBlock()

Specified by:

removeLastBlock in interface ai.djl.nn.SymbolBlock

forward

public ai.djl.ndarray.NDList forward(ai.djl.training.ParameterStore parameterStore,
                                     ai.djl.ndarray.NDList inputs,
                                     boolean training,
                                     ai.djl.util.PairList<java.lang.String,java.lang.Object> params)

Specified by:

forward in interface ai.djl.nn.Block

setInitializer

public void setInitializer(ai.djl.training.initializer.Initializer initializer)

Specified by:

setInitializer in interface ai.djl.nn.Block

setInitializer

public void setInitializer(ai.djl.training.initializer.Initializer initializer,
                           java.lang.String paramName)

Specified by:

setInitializer in interface ai.djl.nn.Block

initialize

public ai.djl.ndarray.types.Shape[] initialize(ai.djl.ndarray.NDManager manager,
                                               ai.djl.ndarray.types.DataType dataType,
                                               ai.djl.ndarray.types.Shape... inputShapes)

Specified by:

initialize in interface ai.djl.nn.Block

isInitialized

public boolean isInitialized()

Specified by:

isInitialized in interface ai.djl.nn.Block

cast

public void cast(ai.djl.ndarray.types.DataType dataType)

Specified by:

cast in interface ai.djl.nn.Block

clear

public void clear()

Specified by:

clear in interface ai.djl.nn.Block

describeInput

public ai.djl.util.PairList<java.lang.String,ai.djl.ndarray.types.Shape> describeInput()

Specified by:

describeInput in interface ai.djl.nn.Block

getChildren

public ai.djl.nn.BlockList getChildren()

Specified by:

getChildren in interface ai.djl.nn.Block

getDirectParameters

public java.util.List<ai.djl.nn.Parameter> getDirectParameters()

Specified by:

getDirectParameters in interface ai.djl.nn.Block

getParameters

public ai.djl.nn.ParameterList getParameters()

Specified by:

getParameters in interface ai.djl.nn.Block

getParameterShape

public ai.djl.ndarray.types.Shape getParameterShape(java.lang.String name,
                                                    ai.djl.ndarray.types.Shape[] inputShapes)

Specified by:

getParameterShape in interface ai.djl.nn.Block

getOutputShapes

public ai.djl.ndarray.types.Shape[] getOutputShapes(ai.djl.ndarray.NDManager manager,
                                                    ai.djl.ndarray.types.Shape[] inputShapes)

Specified by:

getOutputShapes in interface ai.djl.nn.Block

saveParameters

public void saveParameters(java.io.DataOutputStream os)

Specified by:

saveParameters in interface ai.djl.nn.Block

loadParameters

public void loadParameters(ai.djl.ndarray.NDManager manager,
                           java.io.DataInputStream is)

Specified by:

loadParameters in interface ai.djl.nn.Block