libtorch

README

libtorch-go

libtorch Go wrapper for model inference

Dependencies

• libtorch

Usage

package main

import (
	"fmt"

	"changkun.de/x/libtorch-go"
)

func main() {
	m := libtorch.NewModel("model.pt")
	defer m.Close()

	nsamples, nfeatures := 2, 4
	x := []float32{1, 2, 3, 4, 1, 2, 3, 4}

	// Output: [13.449158 13.449158]
	fmt.Println(m.Predict(x, nsamples, nfeatures))
}

For more details, see example folder.

Documentation

Index

• type Model
  • func NewModel(path string) *Model
  • func (m *Model) Close()
  • func (m *Model) Predict(x []float32, nsamples, nfeatures int) []float32

Types

type Model

type Model struct {
	// contains filtered or unexported fields
}

func NewModel

func NewModel(path string) *Model

func (*Model) Close

func (m *Model) Close()

func (*Model) Predict

func (m *Model) Predict(x []float32, nsamples, nfeatures int) []float32