algorithm

Documentation

Index

• func ArgMax(data []float32) int
• func Max(a, b int) int
• func Min(a, b int) int
• func Multinomial(probs []float32) int
• func Product(shape []int) int
• func Relu(x float32) float32
• func ReluDerivative(x float32) float32
• func Softmax(data []float32) []float32
• func TopK(data []float32, k int) ([]float32, []int)

Functions

func ArgMax

func ArgMax(data []float32) int

func Max

func Max(a, b int) int

func Min

func Min(a, b int) int

func Multinomial

func Multinomial(probs []float32) int

func Product

func Product(shape []int) int

func Relu

func Relu(x float32) float32

func ReluDerivative

func ReluDerivative(x float32) float32

func Softmax

func Softmax(data []float32) []float32

func TopK

func TopK(data []float32, k int) ([]float32, []int)