Why Go
- Case Studies
  
  Common problems companies solve with Go
- Use Cases
  
  Stories about how and why companies use Go
- Security
  
  How Go can help keep you secure by default
Learn
Docs
- Effective Go
  
  Tips for writing clear, performant, and idiomatic Go code
- Go User Manual
  
  A complete introduction to building software with Go
- Standard library
  
  Reference documentation for Go's standard library
- Release Notes
  
  Learn what's new in each Go release
- API
  
  Reference documentation for the pkg.go.dev API
Packages
Community
- Recorded Talks
  
  Videos from prior events
- Meetups
  
  Meet other local Go developers
- Conferences
  
  Learn and network with Go developers from around the world
- Go blog
  
  The Go project's official blog.
- Go project
  
  Get help and stay informed from Go
- Get connected

mat

package

Go to main page

Versions in this module

v0

v0.0.6

Aug 14, 2026

v0.0.5

Aug 13, 2026

v0.0.4

Aug 4, 2026

Changes in this version

+ var ErrInvalidProbability = errors.New("probability must be in [0, 1)")

+ func AdamDelta(first, second *Matrix, scale, epsilon float32, out *Matrix) error

+ func AdamFirstMoment(moment, gradient *Matrix, beta float32, out *Matrix) error

+ func AdamSecondMoment(moment, gradient *Matrix, beta float32, out *Matrix) error

+ func AllFiniteAccumulate(input, flag *Matrix) error

+ func Dropout(input *Matrix, probability float32, state RandomState, out *Matrix) error

+ func SelectFinite(candidate, original, flag, out *Matrix) error

+ type RandomState struct

+ Counter uint64

+ Seed uint64

+ StreamID uint64

v0.0.3

Jul 31, 2026

v0.0.2

Jul 31, 2026

Changes in this version

+ func BroadcastTo(input, out *Matrix) error

+ func Mul(left, right, out *Matrix) error

+ func ReduceSumTo(input, out *Matrix) error

+ func ReshapeTo(input, out *Matrix) error

type Context

+ func (c *Context) Stats() Stats

+ type Stats struct

+ ComputeSubmissionCount uint64

+ HostReadBytes uint64

+ HostReadCount uint64

+ HostWriteBytes uint64

+ HostWriteCount uint64

+ LiveMatrixBytes uint64

+ MatrixAllocationCount uint64

+ MatrixReleaseCount uint64

+ PeakLiveMatrixBytes uint64

+ ReadbackSubmissionCount uint64

v0.0.1

Jul 28, 2026

Changes in this version

+ var ErrAliasedOutput = errors.New("out must not alias an input")

+ var ErrBackendUnavailable = errors.New("backend unavailable")

+ var ErrContextMismatch = errors.New("matrices must use the same context")

+ var ErrContextNotInitialized = errors.New("context is not initialized")

+ var ErrContextReleased = errors.New("context is released")

+ var ErrDeviceLimit = errors.New("exceeds device limits")

+ var ErrDimensionMismatch = errors.New("dimension mismatch")

+ var ErrInvalidDimension = errors.New("matrix dimensions must be positive")

+ var ErrInvalidMode = errors.New("invalid context mode")

+ var ErrInvalidState = errors.New("invalid matrix state")

+ var ErrKernelLimit = errors.New("exceeds kernel limits")

+ var ErrLengthMismatch = errors.New("data length mismatch")

+ var ErrNilContext = errors.New("context is nil")

+ var ErrNotInitialized = errors.New("not initialized")

+ var ErrOverflow = errors.New("overflow")

+ var ErrReleased = errors.New("released")

+ func Add(left, right, out *Matrix) error

+ func MatMul(left, right, out *Matrix) error

+ func RMSNorm(input, out *Matrix) error

+ func ReduceMax(input, out *Matrix) error

+ func ReduceSum(input, out *Matrix) error

+ func Scale(input *Matrix, scalar float32, out *Matrix) error

+ func Softmax(input, out *Matrix) error

+ func Transp(input, out *Matrix) error

+ type Context struct

+ func NewContext(modes ...ContextMode) (*Context, error)

+ func (c *Context) Close() error

+ func (c *Context) Mode() ContextMode

+ func (c *Context) Release()

+ func (c *Context) Released() bool

+ type ContextMode uint8

+ const UseAuto

+ const UseCPU

+ const UseGPU

+ func (m ContextMode) String() string

+ type Matrix struct

+ func NewMatrix(ctx *Context, rows, cols int) (*Matrix, error)

+ func (m *Matrix) Close() error

+ func (m *Matrix) Cols() int

+ func (m *Matrix) Len() int

+ func (m *Matrix) Read() ([]float32, error)

+ func (m *Matrix) Release()

+ func (m *Matrix) Released() bool

+ func (m *Matrix) Rows() int

+ func (m *Matrix) Shape() Shape

+ func (m *Matrix) String() string

+ func (m *Matrix) Write(data []float32) error

+ type Shape struct

+ func (s Shape) Cols() int

+ func (s Shape) Len() int

+ func (s Shape) Rows() int

+ func (s Shape) String() string