Documentation
¶
Overview ¶
Package tensor provides type-safe tensor operations for the Born ML framework.
Overview ¶
Tensors are the fundamental data structure in Born. This package provides:
- Generic type-safe tensors (Tensor[T, B])
- NumPy-style broadcasting
- Zero-copy operations where possible
- Device abstraction (CPU, CUDA)
Basic Usage ¶
import (
"github.com/born-ml/born/tensor"
"github.com/born-ml/born/backend/cpu"
)
func main() {
backend := cpu.New()
// Create tensors
x := tensor.Zeros[float32](tensor.Shape{2, 3}, backend)
y := tensor.Ones[float32](tensor.Shape{2, 3}, backend)
// Tensor operations
z := x.Add(y)
result := x.MatMul(y.Transpose())
}
Supported Data Types ¶
The tensor package supports the following data types via the DType constraint:
- float32, float64 (floating-point)
- int32, int64 (signed integers)
- uint8 (unsigned integers, useful for images)
- bool (boolean masks)
Device Support ¶
Tensors can reside on different devices:
- CPU: Pure Go implementation (current)
- CUDA: GPU support (planned for v0.3.0)
Broadcasting ¶
Tensor operations follow NumPy broadcasting rules:
a := tensor.Zeros[float32](tensor.Shape{3, 1}, backend) // (3, 1)
b := tensor.Ones[float32](tensor.Shape{3, 4}, backend) // (3, 4)
c := a.Add(b) // (3, 4)
Memory Management ¶
Tensors use zero-copy operations where possible. The underlying data is reference-counted and automatically freed when no longer needed.
Index ¶
- type Backend
- type DType
- type DataType
- type Device
- type RawTensor
- type Shape
- type Tensor
- func Arange[T DType, B Backend](start, end T, b B) *Tensor[T, B]
- func Eye[T DType, B Backend](n int, b B) *Tensor[T, B]
- func FromSlice[T DType, B Backend](data []T, shape Shape, b B) (*Tensor[T, B], error)
- func Full[T DType, B Backend](shape Shape, value T, b B) *Tensor[T, B]
- func New[T DType, B Backend](raw *RawTensor, b B) *Tensor[T, B]
- func Ones[T DType, B Backend](shape Shape, b B) *Tensor[T, B]
- func Rand[T DType, B Backend](shape Shape, b B) *Tensor[T, B]
- func Randn[T DType, B Backend](shape Shape, b B) *Tensor[T, B]
- func Zeros[T DType, B Backend](shape Shape, b B) *Tensor[T, B]
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Backend ¶
Backend interface for device-specific operations. Implementations: cpu.Backend (current), cuda.Backend (planned).
type DType ¶
DType is a constraint for tensor data types. Supported types: float32, float64, int32, int64, uint8, bool.
type RawTensor ¶
RawTensor is the low-level tensor representation. Most users should use the high-level Tensor type instead.
type Shape ¶
Shape represents the dimensions of a tensor. Example: Shape{2, 3, 4} represents a 3D tensor with dimensions 2×3×4.
func BroadcastShapes ¶
BroadcastShapes computes the broadcast shape for two shapes following NumPy broadcasting rules. Returns the resulting shape and two flags indicating if each operand needs broadcasting.
Example:
resultShape, needsBroadcastA, err := tensor.BroadcastShapes(
tensor.Shape{3, 1},
tensor.Shape{3, 4},
)
// resultShape = [3, 4], needsBroadcastA = true
type Tensor ¶
Tensor is a generic type-safe tensor.
T is the data type (float32, float64, int32, int64, uint8, bool). B is the backend implementation (CPU, CUDA, etc.).
Example:
backend := cpu.New()
x := tensor.Zeros[float32](tensor.Shape{2, 3}, backend)
func Arange ¶
Arange creates a 1D tensor with values from start to end (exclusive).
Example:
backend := cpu.New() x := tensor.Arange[float32](0, 10, backend) // [0, 1, 2, ..., 9]
func Eye ¶
Eye creates a 2D identity matrix.
Example:
backend := cpu.New() identity := tensor.Eye[float32](3, backend) // 3x3 identity matrix
func FromSlice ¶
FromSlice creates a tensor from a Go slice.
Example:
backend := cpu.New()
data := []float32{1, 2, 3, 4, 5, 6}
x, err := tensor.FromSlice(data, tensor.Shape{2, 3}, backend)
func Full ¶
Full creates a tensor filled with a specific value.
Example:
backend := cpu.New()
x := tensor.Full[float32](tensor.Shape{2, 3}, 3.14, backend)
func New ¶
New creates a tensor from a raw tensor.
This is a low-level function. Most users should use creation functions like Zeros, Ones, or FromSlice instead.
func Ones ¶
Ones creates a tensor filled with ones.
Example:
backend := cpu.New()
x := tensor.Ones[float32](tensor.Shape{2, 3}, backend)
func Rand ¶
Rand creates a tensor filled with random values from uniform distribution U(0, 1).
Example:
backend := cpu.New()
x := tensor.Rand[float32](tensor.Shape{2, 3}, backend)
Source Files