tensor

package
v0.0.0-...-27647ab Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Oct 17, 2023 License: BSD-3-Clause, MIT Imports: 1 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type BaseTensor

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

func NewBaseTensor

func NewBaseTensor(
	shape Shape,
) BaseTensor

func (BaseTensor) Sz

func (t BaseTensor) Sz() Dim

type Dim

type Dim = int64

type Shape

type Shape []Dim

func (Shape) At

func (sh Shape) At(i int) Dim

func (Shape) Dim

func (sh Shape) Dim() Dim

func (Shape) Equals

func (sh Shape) Equals(o Shape) bool

type ShapeStride

type ShapeStride struct {
	Shape, Stride Dim
}

type Strides

type Strides []Dim

func BuildStrides

func BuildStrides(shape Shape, width Dim) Strides

func (Strides) Offset

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

type View

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

func NewView

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

Jump to

Keyboard shortcuts

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