go-ml-dev/nn

View on GitHub
mx/mxnet.go

Summary

Maintainability
A
0 mins
Test Coverage
A
100%
package mx

import (
    "go4ml.xyz/base/fu"
    "go4ml.xyz/nn/mx/capi"
)

const (
    VersionMajor = 1
    VersionMinor = 5
    VersionPatch = 0
)

const Version fu.VersionType = VersionMajor*10000 + VersionMinor*100 + VersionPatch

func LibVersion() fu.VersionType {
    return fu.VersionType(capi.LibVersion)
}

func GpuCount() int {
    return capi.GpuCount
}

func RandomSeed(seed int) {
    capi.RandomSeed(seed)
}