Documentation
¶
Index ¶
- func Concat(ts []tensor.Tensor, dim int) (o tensor.Tensor, err error)
- func Eye(d int, withGrad bool) (o tensor.Tensor, err error)
- func Full(dims []int, value float64, withGrad bool) (o tensor.Tensor, err error)
- func Of(data any, withGrad bool) (o tensor.Tensor, err error)
- func Ones(dims []int, withGrad bool) (o tensor.Tensor, err error)
- func RandN(dims []int, u, s float64, withGrad bool) (o tensor.Tensor, err error)
- func RandU(dims []int, l, u float64, withGrad bool) (o tensor.Tensor, err error)
- func Zeros(dims []int, withGrad bool) (o tensor.Tensor, err error)
- type CPUTensor
- func (t *CPUTensor) Add(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Argmax(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Argmin(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) At(index ...int) (value float64, err error)
- func (t *CPUTensor) Avg() (value float64)
- func (t *CPUTensor) AvgAlong(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Broadcast(shape []int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Cos() (o tensor.Tensor)
- func (t *CPUTensor) Cosh() (o tensor.Tensor)
- func (t *CPUTensor) Device() (d tensor.Device)
- func (t *CPUTensor) Div(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Dot(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) ElMax(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) ElMin(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Eq(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Equals(u tensor.Tensor) (are bool, err error)
- func (t *CPUTensor) Exp() (o tensor.Tensor)
- func (t *CPUTensor) Flatten(fromDim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Ge(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) GradContext() (gctx any)
- func (t *CPUTensor) Gradient() (g tensor.Tensor)
- func (t *CPUTensor) GradientTracked() (tracked bool)
- func (t *CPUTensor) Gt(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Le(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Log() (o tensor.Tensor)
- func (t *CPUTensor) Lt(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) MatMul(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Max() (value float64)
- func (t *CPUTensor) MaxAlong(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Mean() (value float64)
- func (t *CPUTensor) MeanAlong(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Min() (value float64)
- func (t *CPUTensor) MinAlong(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Mul(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) NElems() (n int)
- func (t *CPUTensor) Ne(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Patch(index []tensor.Range, u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Pow(u float64) (o tensor.Tensor)
- func (t *CPUTensor) ResetGradContext(tracked bool)
- func (t *CPUTensor) Reshape(shape []int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Scale(u float64) (o tensor.Tensor)
- func (t *CPUTensor) Shape() (shape []int)
- func (t *CPUTensor) Sin() (o tensor.Tensor)
- func (t *CPUTensor) Sinh() (o tensor.Tensor)
- func (t *CPUTensor) Slice(index []tensor.Range) (o tensor.Tensor, err error)
- func (t *CPUTensor) Squeeze(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Std() (value float64)
- func (t *CPUTensor) StdAlong(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Sub(u tensor.Tensor) (o tensor.Tensor, err error)
- func (t *CPUTensor) Sum() (value float64)
- func (t *CPUTensor) SumAlong(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Tan() (o tensor.Tensor)
- func (t *CPUTensor) Tanh() (o tensor.Tensor)
- func (t *CPUTensor) Transpose() (o tensor.Tensor, err error)
- func (t *CPUTensor) UnSqueeze(dim int) (o tensor.Tensor, err error)
- func (t *CPUTensor) Var() (value float64)
- func (t *CPUTensor) VarAlong(dim int) (o tensor.Tensor, err error)
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
Types ¶
type CPUTensor ¶
type CPUTensor struct {
// contains filtered or unexported fields
}
func (*CPUTensor) GradContext ¶
func (*CPUTensor) GradientTracked ¶ added in v0.3.1
func (*CPUTensor) ResetGradContext ¶ added in v0.0.2
Source Files
Click to show internal directories.
Click to hide internal directories.