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

• Constants
• 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
• type OutputSpec
• type SessionOptions
• type TF
  • func New(opts ...Option) (TF, error)

Constants

const (
	TwoDim uint8 = iota + 2
	ThreeDim
)

Types

type Operation

type Operation = tf.Operation

type Option

type Option func(*tensorflow)

func WithExportPath

func WithExportPath(path string) Option

func WithFeed

func WithFeed(operationName string, outputIndex int) Option

func WithFeeds

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

func WithFetch

func WithFetch(operationName string, outputIndex int) Option

func WithFetches

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

func WithNdim

func WithNdim(ndim uint8) 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 OutputSpec

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

type SessionOptions

type SessionOptions = tf.SessionOptions

type TF

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

func New

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