tensorset

Documentation

Index

• type TensorSet
  • func NewTensorSet(data, ans []*tensor.Tensor) *TensorSet
  • func (m *TensorSet) Close()
  • func (m *TensorSet) GetAnswersSize() []int
  • func (m *TensorSet) GetDataSize() []int
  • func (m *TensorSet) GetNextSet() (*tensor.Tensor, *tensor.Tensor, error)
  • func (m *TensorSet) GetSetSize() int
  • func (m *TensorSet) IsAnswer(out *tensor.Tensor, ans *tensor.Tensor) bool
  • func (m *TensorSet) Reset()

Types

type TensorSet

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

TensorSet is just a simple Dataset created from two tensor arrays: the data and the expected answers

func NewTensorSet

func NewTensorSet(data, ans []*tensor.Tensor) *TensorSet

func (*TensorSet) Close

func (m *TensorSet) Close()

func (*TensorSet) GetAnswersSize

func (m *TensorSet) GetAnswersSize() []int

func (*TensorSet) GetDataSize

func (m *TensorSet) GetDataSize() []int

func (*TensorSet) GetNextSet

func (m *TensorSet) GetNextSet() (*tensor.Tensor, *tensor.Tensor, error)

func (*TensorSet) GetSetSize

func (m *TensorSet) GetSetSize() int

func (*TensorSet) IsAnswer

func (m *TensorSet) IsAnswer(out *tensor.Tensor, ans *tensor.Tensor) bool

func (*TensorSet) Reset

func (m *TensorSet) Reset()