Documentation
¶
Index ¶
- type OpInfo
- type Ops
- func (p *Ops) Backward(handle *cudnn.Handler, alpha, beta float64, x, dx, y, dy *tensor.Volume) error
- func (p *Ops) Destroy() error
- func (p *Ops) Forward(handle *cudnn.Handler, alpha, beta float64, x, y *tensor.Volume) error
- func (p *Ops) Info() (OpInfo, error)
- func (p *Ops) OutputDims(input *tensor.Volume) ([]int32, error)
- func (p *Ops) Properties() (gocudnn.PoolingMode, gocudnn.NANProp, []int32, []int32, []int32, error)
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type OpInfo ¶
type OpInfo struct {
Mode gocudnn.PoolingMode `json:"Mode"`
Nan gocudnn.NANProp `json:"NAN"`
Window []int32 `json:"Window"`
Padding []int32 `json:"Padding"`
Stride []int32 `json:"Stride"`
}
OpInfo is contains all the necessarry information to build a pooling Ops.
type Ops ¶
type Ops struct {
// contains filtered or unexported fields
}
Ops holds what the data that is need to perform the pooling operations
func StageOperation ¶
func StageOperation(mode gocudnn.PoolingMode, nan gocudnn.NANProp, window, padding, stride []int32) (*Ops, error)
StageOperation builds the pooling ops
func StageOperationReverse ¶
func StageOperationReverse(mode gocudnn.PoolingMode, nan gocudnn.NANProp, window, padding, stride []int32) (*Ops, error)
StageOperationReverse builds a reverse pooling operation. Experemental!!!!
Will be testing soon. This is just running the backward pooling when forward is called. and it is runing forward pooling when backward is called.
func (*Ops) Backward ¶
func (p *Ops) Backward(handle *cudnn.Handler, alpha, beta float64, x, dx, y, dy *tensor.Volume) error
Backward does the backward propagation operation
func (*Ops) OutputDims ¶
OutputDims returns the dims the output wil have considering the input
Source Files
Click to show internal directories.
Click to hide internal directories.