gorgonia.org/gorgonia@v0.9.17/ops/nn/doc.go (about)

     1  // Package nnops implements some operators that have both a pure go implementation and a cuda implementation
     2  // to use the cuda version, assuming that you have the pre-requisites, simply compile or run the code with the `cuda tag`
     3  //  go run -tags='cuda'
     4  package nnops