Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type ConvolutionLayer ¶
type ConvolutionLayer struct {
// contains filtered or unexported fields
}
func NewConvolutionLayer ¶
func NewConvolutionLayer(filterDimensionSizes []int, depth int, inputDims []int) *ConvolutionLayer
func (*ConvolutionLayer) BackwardPropagation ¶
func (*ConvolutionLayer) ForwardPropagation ¶
func (c *ConvolutionLayer) ForwardPropagation(input maths.Tensor) maths.Tensor
func (*ConvolutionLayer) OutputDims ¶
func (c *ConvolutionLayer) OutputDims() []int
func (*ConvolutionLayer) SaveFiltersAsImages ¶
func (c *ConvolutionLayer) SaveFiltersAsImages(path string) (int, error)
SaveFiltersAsImages saves the filters to images relative to 'path' returns the amount of images saved. saves as grayscale for now
type FullyConnectedLayer ¶
type FullyConnectedLayer struct {
// contains filtered or unexported fields
}
func NewFullyConnectedLayer ¶
func NewFullyConnectedLayer(outputLength int, inputDims []int) *FullyConnectedLayer
func (*FullyConnectedLayer) BackwardPropagation ¶
func (*FullyConnectedLayer) ForwardPropagation ¶
func (d *FullyConnectedLayer) ForwardPropagation(input maths.Tensor) maths.Tensor
func (*FullyConnectedLayer) OutputDims ¶
func (d *FullyConnectedLayer) OutputDims() []int
type MaxPoolingLayer ¶
type MaxPoolingLayer struct {
// contains filtered or unexported fields
}
func NewMaxPoolingLayer ¶
func NewMaxPoolingLayer(strides, sizes, inputDims []int) *MaxPoolingLayer
func (*MaxPoolingLayer) BackwardPropagation ¶
func (*MaxPoolingLayer) ForwardPropagation ¶
func (m *MaxPoolingLayer) ForwardPropagation(input maths.Tensor) maths.Tensor
func (*MaxPoolingLayer) OutputDims ¶
func (m *MaxPoolingLayer) OutputDims() []int
type ReLULayer ¶
type ReLULayer struct {
// contains filtered or unexported fields
}
func NewReLULayer ¶
func (*ReLULayer) BackwardPropagation ¶
func (*ReLULayer) ForwardPropagation ¶
func (*ReLULayer) OutputDims ¶
type SoftmaxLayer ¶
type SoftmaxLayer struct {
// contains filtered or unexported fields
}
func NewSoftmaxLayer ¶
func NewSoftmaxLayer(inputDims []int) *SoftmaxLayer
func (*SoftmaxLayer) BackwardPropagation ¶
func (*SoftmaxLayer) ForwardPropagation ¶
func (o *SoftmaxLayer) ForwardPropagation(input maths.Tensor) maths.Tensor
func (*SoftmaxLayer) OutputDims ¶
func (o *SoftmaxLayer) OutputDims() []int
Source Files
Click to show internal directories.
Click to hide internal directories.