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

tensor

package

Go to main page

Versions in this module

v1

v1.3.1

Jul 21, 2026

v1.3.0

Jul 20, 2026

Changes in this version

+ var UseGPU = false

+ func AddBiasLastDimInGPU(out, bias *Tensor) bool

+ func ClearPersistentGPU()

+ func CopyData(dst, src []float32)

+ func InitGPU() error

+ func ShutdownGPU()

+ type AllocReporter interface

+ ReportAlloc func(t *Tensor)

+ func FirstReporter(rep ...AllocReporter) AllocReporter

+ type Backend interface

+ GemmDeviceAt func(aDev, bDev, cDev cuda.DevicePtr, aOff, bOff, cOff, M, N, K int) error

+ GemmF32 func(A, B, C []float32, M, N, K int) error

+ GemmStridedBatchedAt func(aDev, bDev, cDev cuda.DevicePtr, aOff, bOff, cOff, M, N, K, batchCount int) error

+ Init func() error

+ Shutdown func()

type Tensor

+ func NewGPU(shape []int, data []float32, rep AllocReporter) *Tensor

+ func NewReported(rep AllocReporter, shape ...int) *Tensor

+ func NewWithData(shape []int, data []float32) *Tensor

+ func (a *Tensor) TransposeInto(dst *Tensor)

+ func (t *Tensor) EnsureCPU()

+ func (t *Tensor) EnsureGPU() (cuda.DevicePtr, error)

+ func (t *Tensor) GPUResident() bool

+ func (t *Tensor) MarkCPUModified()

+ func (t *Tensor) MarkGPUModified()

v1.2.0

Jun 29, 2026

Changes in this version

+ var Checks = false

+ var Threads = max(1, runtime.NumCPU() / 3 * 2)

+ type ElemFunc func(i int, v float32) float32

+ type Tensor struct

+ Data []float32

+ Shape []int

+ func New(shape ...int) *Tensor

+ func NewRandom(shape []int, scale float32) *Tensor

+ func UnmarshalBinary(data []byte) (*Tensor, error)

+ func WithData(shape []int, data []float32) *Tensor

+ func (a *Tensor) Add(b *Tensor) *Tensor

+ func (a *Tensor) AddIn(b *Tensor)

+ func (a *Tensor) AddMulNumIn(b *Tensor, n float32)

+ func (a *Tensor) AddNumIn(n float32)

+ func (a *Tensor) AddNumInAt(p []int, n float32)

+ func (a *Tensor) Apply(fn ElemFunc) *Tensor

+ func (a *Tensor) ApplyIn(fn ElemFunc)

+ func (a *Tensor) ArgMax() int

+ func (a *Tensor) CheckData()

+ func (a *Tensor) Clear()

+ func (a *Tensor) Copy() *Tensor

+ func (a *Tensor) Dim() int

+ func (a *Tensor) Exp() *Tensor

+ func (a *Tensor) ExpectShape(shape ...int)

+ func (a *Tensor) Fill(value float32)

+ func (a *Tensor) Free()

+ func (a *Tensor) Get(p ...int) float32

+ func (a *Tensor) Get2D(row, col int) float32

+ func (a *Tensor) IsSameShape(b *Tensor) bool

+ func (a *Tensor) Load(file io.Reader) error

+ func (a *Tensor) Log() *Tensor

+ func (a *Tensor) MarshalBinary() []byte

+ func (a *Tensor) MatMul(b *Tensor) *Tensor

+ func (a *Tensor) MatMulBatch(b *Tensor) *Tensor

+ func (a *Tensor) MatMulLastDim(b *Tensor) *Tensor

+ func (a *Tensor) Max() float32

+ func (a *Tensor) Mean() float32

+ func (a *Tensor) Mul(b *Tensor) *Tensor

+ func (a *Tensor) MulIn(b *Tensor)

+ func (a *Tensor) MulNum(n float32) *Tensor

+ func (a *Tensor) MulNumIn(n float32)

+ func (a *Tensor) Numel() int

+ func (a *Tensor) Permute(axes ...int) *Tensor

+ func (a *Tensor) ReduceToShape(shape []int) *Tensor

+ func (a *Tensor) Reshape(shape ...int) *Tensor

+ func (a *Tensor) Save(file io.Writer) error

+ func (a *Tensor) SelectDim(dim, index int) *Tensor

+ func (a *Tensor) Set(p []int, value float32)

+ func (a *Tensor) Set1D(row int, value float32)

+ func (a *Tensor) Set2D(row, col int, value float32)

+ func (a *Tensor) SetData(b *Tensor)

+ func (a *Tensor) SliceDim(dim, start, end int) *Tensor

+ func (a *Tensor) Softmax() *Tensor

+ func (a *Tensor) Sub(b *Tensor) *Tensor

+ func (a *Tensor) SubIn(b *Tensor)

+ func (a *Tensor) SubNum(n float32) *Tensor

+ func (a *Tensor) Sum() float32

+ func (a *Tensor) SumDim(dim int) *Tensor

+ func (a *Tensor) Transpose() *Tensor

+ func (a *Tensor) View(p ...int) *Tensor

+ func (logits *Tensor) SampleSoftmax(temperature float32) int