tensor

package

v0.0.0-...-27647ab Latest Latest Go to latest Published: Oct 17, 2023 License: BSD-3-Clause, MIT Imports: 1 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

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

func NewBaseTensor(
	shape Shape,
) BaseTensor

func (t BaseTensor) Sz() Dim

type Dim = int64

type Shape []Dim

func (sh Shape) At(i int) Dim

func (sh Shape) Dim() Dim

func (sh Shape) Equals(o Shape) bool

type ShapeStride struct {
	Shape, Stride Dim
}

type Strides []Dim

func BuildStrides(shape Shape, width Dim) Strides

func (st Strides) Offset(idxs ...Dim) Dim

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

func NewView(shape Shape, strides Strides, offset Dim) View