Versions in this module Expand all Collapse all v0 v0.17.17 Aug 18, 2026 Changes in this version + func Available() bool + type Array struct + func Abs(a *Array, s *Stream) (*Array, error) + func Add(a, b *Array, s *Stream) (*Array, error) + func Arange(start, stop, step float64, dtype Dtype, s *Stream) (*Array, error) + func Cast(a *Array, dtype Dtype, s *Stream) (*Array, error) + func Concatenate(arrays []*Array, s *Stream) (*Array, error) + func ConcatenateAxis(arrays []*Array, axis int, s *Stream) (*Array, error) + func Cos(a *Array, s *Stream) (*Array, error) + func Divide(a, b *Array, s *Stream) (*Array, error) + func Exp(a *Array, s *Stream) (*Array, error) + func Gather(a, indices *Array, axes, sliceSizes []int, s *Stream) (*Array, error) + func GatherAxis(a, indices *Array, axis int, sliceSizes []int, s *Stream) (*Array, error) + func MatMul(a, b *Array, s *Stream) (*Array, error) + func Max(a *Array, axes []int, keepdims bool, s *Stream) (*Array, error) + func Maximum(a, b *Array, s *Stream) (*Array, error) + func Mean(a *Array, axes []int, keepdims bool, s *Stream) (*Array, error) + func Multiply(a, b *Array, s *Stream) (*Array, error) + func NewArrayFromFloat32(data []float32, shape []int) (*Array, error) + func NewArrayFromInt32(data []int32, shape []int) (*Array, error) + func NewArrayFromInt64(data []int64, shape []int) (*Array, error) + func Pad(a *Array, axes, low, high []int, padValue *Array, s *Stream) (*Array, error) + func Power(a *Array, exp float32, s *Stream) (*Array, error) + func RepeatAxis(a *Array, repeats, axis int, s *Stream) (*Array, error) + func Reshape(a *Array, shape []int, s *Stream) (*Array, error) + func Sin(a *Array, s *Stream) (*Array, error) + func Slice(a *Array, start, stop, strides []int, s *Stream) (*Array, error) + func Softmax(a *Array, s *Stream) (*Array, error) + func SoftmaxAxis(a *Array, axis int, s *Stream) (*Array, error) + func Split(a *Array, indices []int, s *Stream) ([]*Array, error) + func SplitAxis(a *Array, indices []int, axis int, s *Stream) ([]*Array, error) + func Sqrt(a *Array, s *Stream) (*Array, error) + func Square(a *Array, s *Stream) (*Array, error) + func SqueezeAxis(a *Array, axis int, s *Stream) (*Array, error) + func Subtract(a, b *Array, s *Stream) (*Array, error) + func Sum(a *Array, axes []int, keepdims bool, s *Stream) (*Array, error) + func Tanh(a *Array, s *Stream) (*Array, error) + func Transpose(a *Array, s *Stream) (*Array, error) + func TransposeAxes(a *Array, axes []int, s *Stream) (*Array, error) + func Tril(a *Array, k int, s *Stream) (*Array, error) + func Where(condition, x, y *Array, s *Stream) (*Array, error) + func (a *Array) Dtype() Dtype + func (a *Array) Eval() error + func (a *Array) Float32Data() ([]float32, error) + func (a *Array) Free() + func (a *Array) Int64Data() ([]int64, error) + func (a *Array) Ndim() int + func (a *Array) Shape() []int + func (a *Array) Size() int + type Dtype int + const BFloat16 + const Bool + const Complex64 + const Float16 + const Float32 + const Float64 + const Int16 + const Int32 + const Int64 + const Int8 + const UInt16 + const UInt32 + const UInt64 + const UInt8 + type Stream struct + func DefaultGPUStream() (*Stream, error) + func DefaultStream() (*Stream, error) + func (s *Stream) Free() + func (s *Stream) Synchronize() error
Go to main page