tensor

package module

v0.0.0-...-e2bb1e6 Latest Latest Go to latest Published: Aug 29, 2025 License: MIT Imports: 9 Imported by: 1

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/olimci/autograd

Links

Open Source Insights

Documentation ¶

Index ¶

type Float
type IterMulti
- func NewIterMulti[F Float](tensors ...*Tensor[F]) *IterMulti
type IterND
type IterStride
type Tensor

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

This section is empty.

Types ¶

type Float ¶

type Float interface {
	~float32 | ~float64
}

type IterMulti ¶

type IterMulti struct {
	// contains filtered or unexported fields
}

func NewIterMulti ¶

func NewIterMulti[F Float](tensors ...*Tensor[F]) *IterMulti

func (*IterMulti) Done ¶

func (it *IterMulti) Done() bool

func (*IterMulti) Index ¶

func (it *IterMulti) Index() []int

func (*IterMulti) Next ¶

func (it *IterMulti) Next()

func (*IterMulti) Offsets ¶

func (it *IterMulti) Offsets() []int

type IterND ¶

type IterND struct {
	// contains filtered or unexported fields
}

func (*IterND) Done ¶

func (it *IterND) Done() bool

func (*IterND) Index ¶

func (it *IterND) Index() []int

func (*IterND) Next ¶

func (it *IterND) Next()

func (*IterND) Offset ¶

func (it *IterND) Offset() int

type IterStride ¶

type IterStride struct {
	// contains filtered or unexported fields
}

func (*IterStride) Count ¶

func (it *IterStride) Count() int

func (*IterStride) Done ¶

func (it *IterStride) Done() bool

func (*IterStride) Index ¶

func (it *IterStride) Index() []int

func (*IterStride) Next ¶

func (it *IterStride) Next()

func (*IterStride) Offset ¶

func (it *IterStride) Offset() int

type Tensor ¶

type Tensor[F Float] struct {
	Data   []F
	Shape  []int
	Stride []int
}

func Add ¶

func Add[F Float](tensors ...*Tensor[F]) *Tensor[F]

func AddS ¶

func AddS[F Float](t *Tensor[F], f F) *Tensor[F]

func Apply ¶

func Apply[F Float](t *Tensor[F], fn func(F) F) *Tensor[F]

func Apply2 ¶

func Apply2[F Float](a, b *Tensor[F], fn func(F, F) F) *Tensor[F]

func Conv ¶

func Conv[F Float](
	x, w *Tensor[F],
	bias *Tensor[F],
	sH, sW, pH, pW int,
) *Tensor[F]

func Div ¶

func Div[F Float](a, b *Tensor[F]) *Tensor[F]

func DivS ¶

func DivS[F Float](t *Tensor[F], f F) *Tensor[F]

func FromGen ¶

func FromGen[F Float](gen func(i ...int) F, shape ...int) *Tensor[F]

func FromGen0 ¶

func FromGen0[F Float](gen func() F, shape ...int) *Tensor[F]

func FromGen1 ¶

func FromGen1[F Float](gen func(i int) F, shape ...int) *Tensor[F]

func Fs ¶

func Fs[F Float](f F, shape ...int) *Tensor[F]

func FsLike ¶

func FsLike[F Float](f F, t *Tensor[F]) *Tensor[F]

func MatMul ¶

func MatMul[F Float](a, b *Tensor[F]) *Tensor[F]

func Mul ¶

func Mul[F Float](tensors ...*Tensor[F]) *Tensor[F]

func MulS ¶

func MulS[F Float](t *Tensor[F], f F) *Tensor[F]

func Neg ¶

func Neg[F Float](t *Tensor[F]) *Tensor[F]

func New ¶

func New[F Float](data []F, shape ...int) *Tensor[F]

func Ones ¶

func Ones[F Float](shape ...int) *Tensor[F]

func OnesLike ¶

func OnesLike[F Float](t *Tensor[F]) *Tensor[F]

func Rand ¶

func Rand[F Float](shape ...int) *Tensor[F]

func RandLike ¶

func RandLike[F Float](t *Tensor[F]) *Tensor[F]

func Sub ¶

func Sub[F Float](a, b *Tensor[F]) *Tensor[F]

func SubS ¶

func SubS[F Float](t *Tensor[F], f F) *Tensor[F]

func Zeros ¶

func Zeros[F Float](shape ...int) *Tensor[F]

func ZerosLike ¶

func ZerosLike[F Float](t *Tensor[F]) *Tensor[F]

func (*Tensor[F]) Add ¶

func (t *Tensor[F]) Add(tensors ...*Tensor[F]) *Tensor[F]

func (*Tensor[F]) AddS ¶

func (t *Tensor[F]) AddS(f F) *Tensor[F]

func (*Tensor[F]) Apply ¶

func (t *Tensor[F]) Apply(fn func(F) F) *Tensor[F]

func (*Tensor[F]) Apply2 ¶

func (t *Tensor[F]) Apply2(s *Tensor[F], fn func(F, F) F) *Tensor[F]

func (*Tensor[F]) ArgMax ¶

func (t *Tensor[F]) ArgMax() (F, []int)

func (*Tensor[F]) ArgMin ¶

func (t *Tensor[F]) ArgMin() (F, []int)

func (*Tensor[F]) Clone ¶

func (t *Tensor[F]) Clone() *Tensor[F]

func (*Tensor[F]) Copy ¶

func (t *Tensor[F]) Copy(to *Tensor[F]) *Tensor[F]

func (*Tensor[F]) Div ¶

func (t *Tensor[F]) Div(s *Tensor[F]) *Tensor[F]

func (*Tensor[F]) DivS ¶

func (t *Tensor[F]) DivS(f F) *Tensor[F]

func (*Tensor[F]) Flatten ¶

func (t *Tensor[F]) Flatten() *Tensor[F]

func (*Tensor[F]) Get ¶

func (t *Tensor[F]) Get(index ...int) F

func (*Tensor[F]) IterND ¶

func (t *Tensor[F]) IterND() *IterND

func (*Tensor[F]) IterStride ¶

func (t *Tensor[F]) IterStride() *IterStride

func (*Tensor[F]) MarshalJSON ¶

func (t *Tensor[F]) MarshalJSON() ([]byte, error)

func (*Tensor[F]) MaskGE ¶

func (t *Tensor[F]) MaskGE(f F) *Tensor[F]

func (*Tensor[F]) MaskLE ¶

func (t *Tensor[F]) MaskLE(f F) *Tensor[F]

func (*Tensor[F]) MatMul ¶

func (t *Tensor[F]) MatMul(a, b *Tensor[F]) *Tensor[F]

func (*Tensor[F]) Max ¶

func (t *Tensor[F]) Max() F

func (*Tensor[F]) Mean ¶

func (t *Tensor[F]) Mean() F

func (*Tensor[F]) Min ¶

func (t *Tensor[F]) Min() F

func (*Tensor[F]) Mul ¶

func (t *Tensor[F]) Mul(tensors ...*Tensor[F]) *Tensor[F]

func (*Tensor[F]) MulS ¶

func (t *Tensor[F]) MulS(f F) *Tensor[F]

func (*Tensor[F]) Neg ¶

func (t *Tensor[F]) Neg() *Tensor[F]

func (*Tensor[F]) Pad ¶

func (t *Tensor[F]) Pad(pads [][2]int, with F) *Tensor[F]

func (*Tensor[F]) Pad2D ¶

func (t *Tensor[F]) Pad2D(u, d, l, r int, with F) *Tensor[F]

func (*Tensor[F]) Permute ¶

func (t *Tensor[F]) Permute(axes ...int) *Tensor[F]

func (*Tensor[F]) Prod ¶

func (t *Tensor[F]) Prod() F

func (*Tensor[F]) Reshape ¶

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

func (*Tensor[F]) ReverseAxes ¶

func (t *Tensor[F]) ReverseAxes(axes ...int) *Tensor[F]

func (*Tensor[F]) Set ¶

func (t *Tensor[F]) Set(f F, index ...int)

func (*Tensor[F]) Size ¶

func (t *Tensor[F]) Size() int

func (*Tensor[F]) String ¶

func (t *Tensor[F]) String() string

func (*Tensor[F]) Sub ¶

func (t *Tensor[F]) Sub(s *Tensor[F]) *Tensor[F]

func (*Tensor[F]) SubS ¶

func (t *Tensor[F]) SubS(f F) *Tensor[F]

func (*Tensor[F]) Sum ¶

func (t *Tensor[F]) Sum() F

func (*Tensor[F]) SumAxes ¶

func (t *Tensor[F]) SumAxes(axes ...int) *Tensor[F]

func (*Tensor[F]) T ¶

func (t *Tensor[F]) T() *Tensor[F]

func (*Tensor[F]) UnmarshalJSON ¶

func (t *Tensor[F]) UnmarshalJSON(b []byte) error

func (*Tensor[F]) Var ¶

func (t *Tensor[F]) Var() F

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL