pytorch

README

go-pytorch

Documentation

Index

• Constants
• Variables
• func SetUseCPU()
• func SetUseGPU()
• type Predictor
  • func New(ctx context.Context, opts ...options.Option) (*Predictor, error)
  • func (p *Predictor) Close()
  • func (p *Predictor) Predict(ctx context.Context, data []float32) error
  • func (p *Predictor) ReadPredictionOutput(ctx context.Context) ([]float32, error)

Constants

const (
	CPUMode = 0
	GPUMode = 1
)

Variables

var (
	Version   = "0.0.1"
	BuildDate = "undefined"
	GitCommit = "undefined"
)

Functions

func SetUseCPU

func SetUseCPU()

func SetUseGPU

func SetUseGPU()

Types

type Predictor

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

func New

func New(ctx context.Context, opts ...options.Option) (*Predictor, error)

func (*Predictor) Close

func (p *Predictor) Close()

func (*Predictor) Predict

func (p *Predictor) Predict(ctx context.Context, data []float32) error

func (*Predictor) ReadPredictionOutput

func (p *Predictor) ReadPredictionOutput(ctx context.Context) ([]float32, error)