Documentation ¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Config ¶
type Config struct { KernelSizeX int KernelSizeY int XStride int YStride int InputChannels int OutputChannels int Mask []int Activation ag.OpName }
Config provides configuration settings for a convolution Model.
type Model ¶
type Model struct { nn.BaseModel Config Config K []nn.Param `spago:"type:weights"` B []nn.Param `spago:"type:biases"` // whether to enable the concurrent forward computation on the output channel Concurrent bool `spago:"scope:processor"` }
Model contains the serializable parameters for a convolutional neural network model.
func (*Model) Forward ¶ added in v0.2.0
Forward performs the forward step for each input node and returns the result.
func (*Model) SetConcurrentComputations ¶ added in v0.2.0
SetConcurrentComputations enables or disables the usage of concurrency in the Forward method.
Click to show internal directories.
Click to hide internal directories.