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 Closer
• type Operation
• type Option
  • func WithExportPath(path string) Option
  • func WithFeed(operationName string, outputIndex int) Option
  • func WithFeeds(operationNames []string, outputIndexes []int) Option
  • func WithFetch(operationName string, outputIndex int) Option
  • func WithFetches(operationNames []string, outputIndexes []int) Option
  • func WithNdim(ndim uint8) 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
  • func WithWarmupInputs(warmupInputs ...string) Option
• type OutputSpec
• type SessionOptions
• type TF
  • func New(opts ...Option) (TF, error)

Types

type Closer

type Closer = io.Closer

Closer is a type alias io.Closer.

type Operation

type Operation = tf.Operation

Operation is a type alias for tensorflow.Operation.

type Option

type Option func(*tensorflow)

Option is tensorflow configure.

func WithExportPath

func WithExportPath(path string) Option

WithExportPath returns Option that sets exportDir.

func WithFeed

func WithFeed(operationName string, outputIndex int) Option

WithFeed returns Option that sets feeds.

func WithFeeds

func WithFeeds(operationNames []string, outputIndexes []int) Option

WithFeeds returns Option that sets feeds.

func WithFetch

func WithFetch(operationName string, outputIndex int) Option

WithFetch returns Option that sets fetches.

func WithFetches

func WithFetches(operationNames []string, outputIndexes []int) Option

WithFetches returns Option that sets fetches.

func WithNdim

func WithNdim(ndim uint8) Option

WithNdim returns Option that sets ndim.

func WithOperations

func WithOperations(opes ...*Operation) Option

WithOperations returns Option that sets operations.

func WithSessionConfig

func WithSessionConfig(cfg []byte) Option

WithSessionConfig returns Option that sets config.

func WithSessionOptions

func WithSessionOptions(opts *SessionOptions) Option

WithSessionOptions returns Option that sets options.

func WithSessionTarget

func WithSessionTarget(tgt string) Option

WithSessionTarget returns Option that sets target.

func WithTags

func WithTags(tags ...string) Option

WithTags returns Option that sets tags.

func WithWarmupInputs

func WithWarmupInputs(warmupInputs ...string) Option

WithWarmupInputs returns Option that sets warmupInputs.

type OutputSpec

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

OutputSpec is the specification of an feed/fetch.

type SessionOptions

type SessionOptions = tf.SessionOptions

SessionOptions is a type alias for tensorflow.SessionOptions.

type TF

type TF interface {
	GetVector(inputs ...string) ([]float64, error)
	GetValue(inputs ...string) (interface{}, error)
	GetValues(inputs ...string) (values []interface{}, err error)
	Closer
}

TF represents a tensorflow interface.

func New

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

New load a tensorlfow model and returns a new tensorflow struct.