tensorflow

Documentation

Overview

Package tensorflow provides implementation of Go API for extract data to vector

Package tensorflow provides implementation of Go API for extract data to vector

Index

• type Feed
• type Fetch
• type Operation
• type Option
  • func WithExportPath(path string) Option
  • func WithOperations(opes ...*Operation) Option
  • func WithSessionConfig(cfg []byte) Option
  • func WithSessionOptions(opts *SessionOptions) Option
  • func WithSessionTarget(tgt string) Option
  • func WithTags(tags ...string) Option
• type SessionOptions
• type TF
  • func New(opts ...Option) (TF, error)

Types

type Feed

type Feed struct {
	InputBytes    []byte
	OperationName string
	OutputIndex   int
}

type Fetch

type Fetch struct {
	OperationName string
	OutputIndex   int
}

type Operation

type Operation = tf.Operation

type Option

type Option func(*tensorflow)

func WithExportPath

func WithExportPath(path string) Option

func WithOperations

func WithOperations(opes ...*Operation) Option

func WithSessionConfig

func WithSessionConfig(cfg []byte) Option

func WithSessionOptions

func WithSessionOptions(opts *SessionOptions) Option

func WithSessionTarget

func WithSessionTarget(tgt string) Option

func WithTags

func WithTags(tags ...string) Option

type SessionOptions

type SessionOptions = tf.SessionOptions

type TF

type TF interface {
	GetVector(feeds []Feed, fetches []Fetch, targets ...*Operation) (values [][][]float64, err error)
	Close() error
}

func New

func New(opts ...Option) (TF, error)