gorgonia: gorgonia.org/gorgonia/ops/nn Index | Files

package nnops

import "gorgonia.org/gorgonia/ops/nn"

Index

Package Files

api_nocuda.go scratch.go utils.go

func BatchNorm Uses

func BatchNorm(x, scale, bias *G.Node, momentum, epsilon float64) (retVal, γ, β *G.Node, op *G.BatchNormOp, err error)

func CheckConvolutionParams Uses

func CheckConvolutionParams(pad, stride, dilation []int) error

func Conv1d Uses

func Conv1d(in, filter *G.Node, kernel, pad, stride, dilation int) (*G.Node, error)

func Conv2d Uses

func Conv2d(im, filter *G.Node, kernelShape tensor.Shape, pad, stride, dilation []int) (retVal *G.Node, err error)

func Dropout Uses

func Dropout(x *G.Node, prob float64) (retVal *G.Node, err error)

func MaxPool2D Uses

func MaxPool2D(x *G.Node, kernel tensor.Shape, pad, stride []int) (*G.Node, error)

func Rectify Uses

func Rectify(x *G.Node) (retVal *G.Node, err error)

Package nnops imports 8 packages (graph) and is imported by 2 packages. Updated 2019-05-22. Refresh now. Tools for package owners.