tensor

Documentation

Index

• func AssertTensorFormat(t Tensor) error
• func BuildTensor(dimension int, numAxes int) interface{}
• func Convert(val interface{}, typ reflect.Type) (interface{}, error)
• func VerticalVector(a Tensor, pos int) []float64
• type Tensor
  • func AddScalar(a Tensor, scalar float64) Tensor
  • func Dot(a, b Tensor) Tensor
  • func DotScalar(a Tensor, scalar float64) Tensor
  • func NewTensor(data [][]float64) Tensor
  • func Pow(a Tensor, pow float64) Tensor
  • func Rand(rows, cols int) Tensor
  • func ReLU(a Tensor) Tensor
  • func ScalarMinusTensor(a Tensor, scalar float64) Tensor
  • func Softmax(tensor Tensor) Tensor
  • func Sub(a, b Tensor) Tensor
  • func Sum(params ...Tensor) Tensor
  • func Tanh(a Tensor) Tensor
  • func TensorOpsScalar(a Tensor, b float64, ops string) Tensor
  • func TensorOpsTensor(a Tensor, b Tensor, ops string) Tensor
  • func TensorOpsTensorWithBroadcasting(a Tensor, b Tensor, ops string) Tensor
  • func Zeros(rows, cols int) Tensor
  • func (t Tensor) IsEqualTo(tb Tensor) bool
  • func (t *Tensor) Reshape(shape []int) Tensor
  • func (t Tensor) SameTensorRowColShape(tb Tensor) error
  • func (t Tensor) SameTensorShape(tb Tensor) error
  • func (t Tensor) Shape() []int
  • func (t Tensor) Transpose() Tensor

Functions

func AssertTensorFormat

func AssertTensorFormat(t Tensor) error

func BuildTensor

func BuildTensor(dimension int, numAxes int) interface{}

func Convert

func Convert(val interface{}, typ reflect.Type) (interface{}, error)

func VerticalVector

func VerticalVector(a Tensor, pos int) []float64

Types

type Tensor

type Tensor struct {
	Rows, Cols   int
	Data         [][]float64
	RequiresGrad bool
}

func AddScalar

func AddScalar(a Tensor, scalar float64) Tensor

func Dot

func Dot(a, b Tensor) Tensor

func DotScalar

func DotScalar(a Tensor, scalar float64) Tensor

func NewTensor

func NewTensor(data [][]float64) Tensor

func Pow

func Pow(a Tensor, pow float64) Tensor

func Rand

func Rand(rows, cols int) Tensor

func ReLU

func ReLU(a Tensor) Tensor

func ScalarMinusTensor

func ScalarMinusTensor(a Tensor, scalar float64) Tensor

func Softmax

func Softmax(tensor Tensor) Tensor

func Sub

func Sub(a, b Tensor) Tensor

func Sum

func Sum(params ...Tensor) Tensor

func Tanh

func Tanh(a Tensor) Tensor

func TensorOpsScalar

func TensorOpsScalar(a Tensor, b float64, ops string) Tensor

func TensorOpsTensor

func TensorOpsTensor(a Tensor, b Tensor, ops string) Tensor

func TensorOpsTensorWithBroadcasting

func TensorOpsTensorWithBroadcasting(a Tensor, b Tensor, ops string) Tensor

func Zeros

func Zeros(rows, cols int) Tensor

func (Tensor) IsEqualTo

func (t Tensor) IsEqualTo(tb Tensor) bool

func (*Tensor) Reshape

func (t *Tensor) Reshape(shape []int) Tensor

func (Tensor) SameTensorRowColShape

func (t Tensor) SameTensorRowColShape(tb Tensor) error

func (Tensor) SameTensorShape

func (t Tensor) SameTensorShape(tb Tensor) error

func (Tensor) Shape

func (t Tensor) Shape() []int

func (Tensor) Transpose

func (t Tensor) Transpose() Tensor