tensor

package
v1.7.3 Latest Latest
Warning

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

 Go to latest Published: Apr 8, 2024 License: MIT Imports: 12 Imported by: 7

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func ClipGradNorm added in v1.2.4 

func ClipGradNorm(params []*Tensor, max, t float64)

func SVD added in v1.6.1 

func SVD(t *Tensor) (*Tensor, *Tensor, *Tensor)

func WriteLeaks added in v1.7.0 

func WriteLeaks(w io.Writer)

Types

type Conv1DOptFunc added in v1.3.7 

type Conv1DOptFunc func(*conv1DOpt)

func Conv1DDilation added in v1.3.7 

func Conv1DDilation(v int) Conv1DOptFunc

func Conv1DGroups added in v1.3.7 

func Conv1DGroups(v int) Conv1DOptFunc

func Conv1DPadding added in v1.3.7 

func Conv1DPadding(v int) Conv1DOptFunc

func Conv1DStride added in v1.3.7 

func Conv1DStride(v int) Conv1DOptFunc

type Conv2DOptFunc added in v1.3.7 

type Conv2DOptFunc func(*conv2DOpt)

func Conv2DDilation added in v1.3.7 

func Conv2DDilation(v int) Conv2DOptFunc

func Conv2DGroups added in v1.3.7 

func Conv2DGroups(v int) Conv2DOptFunc

func Conv2DPadding added in v1.3.7 

func Conv2DPadding(v1, v2 int) Conv2DOptFunc

func Conv2DStride added in v1.3.7 

func Conv2DStride(v1, v2 int) Conv2DOptFunc

type Conv3DOptFunc added in v1.3.7 

type Conv3DOptFunc func(*conv3DOpt)

func Conv3DDilation added in v1.3.7 

func Conv3DDilation(v int) Conv3DOptFunc

func Conv3DGroups added in v1.3.7 

func Conv3DGroups(v int) Conv3DOptFunc

func Conv3DPadding added in v1.3.7 

func Conv3DPadding(v1, v2, v3 int) Conv3DOptFunc

func Conv3DStride added in v1.3.7 

func Conv3DStride(v1, v2, v3 int) Conv3DOptFunc

type Option added in v1.2.0 

type Option func(*options)

func WithDevice added in v1.2.0 

func WithDevice(device consts.DeviceType) Option

func WithShapes added in v1.2.0 

func WithShapes(shapes ...int64) Option

type PoolOpt added in v1.3.0 

type PoolOpt func(*poolOpt)

PoolOpt is an option for pooling operations.

func PoolCeil added in v1.3.0 

func PoolCeil(c bool) PoolOpt

PoolCeil sets the ceil for pooling operations.

func PoolDilation added in v1.3.0 

func PoolDilation(d int) PoolOpt

PoolDilation sets the dilation for pooling operations.

func PoolPadding added in v1.3.0 

func PoolPadding(p int) PoolOpt

PoolPadding sets the padding for pooling operations.

func PoolStride added in v1.3.0 

func PoolStride(s int) PoolOpt

PoolStride sets the stride for pooling operations.

type Tensor 

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

func ARange 

func ARange(n int, dtype consts.ScalarType, opts ...Option) *Tensor

func Cat added in v1.3.2 

func Cat(tensors []*Tensor, dim int) *Tensor

func Embedding added in v1.3.5 

func Embedding(input *Tensor, weight *Tensor, paddingIdx int64) *Tensor

func FromBFloat16 added in v1.5.3 

func FromBFloat16(data []float32, opts ...Option) *Tensor

func FromBFloat16Raw added in v1.5.4 

func FromBFloat16Raw(data []uint16, opts ...Option) *Tensor

func FromBool 

func FromBool(data []bool, opts ...Option) *Tensor

func FromFloat32 

func FromFloat32(data []float32, opts ...Option) *Tensor

func FromFloat64 

func FromFloat64(data []float64, opts ...Option) *Tensor

func FromHalf added in v1.5.3 

func FromHalf(data []float32, opts ...Option) *Tensor

func FromHalfRaw added in v1.5.4 

func FromHalfRaw(data []uint16, opts ...Option) *Tensor

func FromInt16 

func FromInt16(data []int16, opts ...Option) *Tensor

func FromInt32 

func FromInt32(data []int32, opts ...Option) *Tensor

func FromInt64 

func FromInt64(data []int64, opts ...Option) *Tensor

func FromInt8 

func FromInt8(data []int8, opts ...Option) *Tensor

func FromUint8 

func FromUint8(data []uint8, opts ...Option) *Tensor

func HStack 

func HStack(a, b *Tensor) *Tensor

func New added in v1.4.0 

func New(t torch.Tensor) *Tensor

func Outer added in v1.6.3 

func Outer(a, b *Tensor) *Tensor

func Polar added in v1.6.3 

func Polar(abs, angle *Tensor) *Tensor

func ScaledDotProductAttention added in v1.2.3 

func ScaledDotProductAttention(q, k, v, mask *Tensor, drouput float64, isCausal bool) *Tensor

func VStack 

func VStack(a, b *Tensor) *Tensor

func Zeros 

func Zeros(dtype consts.ScalarType, opts ...Option) *Tensor

func (*Tensor) Abs 

func (t *Tensor) Abs() *Tensor

func (*Tensor) Add 

func (t *Tensor) Add(t2 *Tensor) *Tensor

func (*Tensor) AvgPool1D added in v1.3.0 

func (t *Tensor) AvgPool1D(kernel int, opt ...PoolOpt) *Tensor

AvgPool1D returns a new tensor with the result of applying a 1D average pooling operation on the input tensor. kernel: kernel size stride: stride, default 1 padding: padding, default 0 dilation: dilation, default 1 ceil: ceil, default false

func (*Tensor) AvgPool2D added in v1.3.0 

func (t *Tensor) AvgPool2D(kernel int, opt ...PoolOpt) *Tensor

AvgPool2D returns a new tensor with the result of applying a 2D average pooling operation on the input tensor. kernel: kernel size stride: stride, default 1 padding: padding, default 0 dilation: dilation, default 1 ceil: ceil, default false

func (*Tensor) AvgPool3D added in v1.3.0 

func (t *Tensor) AvgPool3D(kernel int, opt ...PoolOpt) *Tensor

AvgPool3D returns a new tensor with the result of applying a 3D average pooling operation on the input tensor. kernel: kernel size stride: stride, default 1 padding: padding, default 0 dilation: dilation, default 1 ceil: ceil, default false

func (*Tensor) BFloat16Raw added in v1.5.4 

func (t *Tensor) BFloat16Raw() []uint16

func (*Tensor) BFloat16Value added in v1.5.3 

func (t *Tensor) BFloat16Value() []float32

func (*Tensor) Backward 

func (t *Tensor) Backward()

func (*Tensor) BackwardRetained 

func (t *Tensor) BackwardRetained()

func (*Tensor) BoolValue 

func (t *Tensor) BoolValue() []bool

func (*Tensor) Contiguous added in v1.3.2 

func (t *Tensor) Contiguous() *Tensor

func (*Tensor) Conv1D added in v1.3.0 

func (t *Tensor) Conv1D(weight, bias *Tensor, opts ...Conv1DOptFunc) *Tensor

Conv1D 1D convolution weight: [out_channels, in_channels/groups, kernel_size] bias: [out_channels] stride: stride, default 1 padding: padding, default 0 dilation: dilation, default 1 groups: groups, default 1

func (*Tensor) Conv2D added in v1.3.0 

func (t *Tensor) Conv2D(weight, bias *Tensor, opts ...Conv2DOptFunc) *Tensor

Conv2D 2D convolution weight: [out_channels, in_channels/groups, kernel_size, kernel_size] bias: [out_channels] stride: stride, default (1, 1) padding: padding, default (0, 0) dilation: dilation, default 1 groups: groups, default 1

func (*Tensor) Conv3D added in v1.3.0 

func (t *Tensor) Conv3D(weight, bias *Tensor, opts ...Conv3DOptFunc) *Tensor

Conv3D 3D convolution weight: [out_channels, in_channels/groups, kernel_size, kernel_size, kernel_size] bias: [out_channels] stride: stride, default (1, 1, 1) padding: padding, default (0, 0, 0) dilation: dilation, default 1 groups: groups, default 1

func (*Tensor) Created added in v1.7.0 

func (t *Tensor) Created() time.Time

func (*Tensor) DeviceType added in v1.4.1 

func (t *Tensor) DeviceType() consts.DeviceType

func (*Tensor) Dims 

func (t *Tensor) Dims() int64

func (*Tensor) Div 

func (t *Tensor) Div(t2 *Tensor) *Tensor

func (*Tensor) Dropout 

func (t *Tensor) Dropout(p float64, train bool) *Tensor

func (*Tensor) ElemCount 

func (t *Tensor) ElemCount() int64

func (*Tensor) ElemSize 

func (t *Tensor) ElemSize() int64

func (*Tensor) Exp 

func (t *Tensor) Exp() *Tensor

func (*Tensor) Expand added in v1.3.2 

func (t *Tensor) Expand(sizes ...int64) *Tensor

func (*Tensor) Flatten added in v1.7.2 

func (t *Tensor) Flatten(startDim, endDim int64) *Tensor

func (*Tensor) Float32Value 

func (t *Tensor) Float32Value() []float32

func (*Tensor) Float64Value 

func (t *Tensor) Float64Value() []float64

func (*Tensor) Gelu added in v1.1.1 

func (t *Tensor) Gelu(tanh bool) *Tensor

func (*Tensor) HalfRaw added in v1.5.4 

func (t *Tensor) HalfRaw() []uint16

func (*Tensor) HalfValue added in v1.5.3 

func (t *Tensor) HalfValue() []float32

func (*Tensor) Int16Value 

func (t *Tensor) Int16Value() []int16

func (*Tensor) Int32Value 

func (t *Tensor) Int32Value() []int32

func (*Tensor) Int64Value 

func (t *Tensor) Int64Value() []int64

func (*Tensor) Int8Value 

func (t *Tensor) Int8Value() []int8

func (*Tensor) LeakyRelu added in v1.6.2 

func (t *Tensor) LeakyRelu(negSlope float64) *Tensor

func (*Tensor) Log 

func (t *Tensor) Log() *Tensor

func (*Tensor) MatMul 

func (t *Tensor) MatMul(t2 *Tensor) *Tensor

func (*Tensor) Max 

func (t *Tensor) Max(dim int64, keepdim bool) *Tensor

func (*Tensor) MaxPool1D added in v1.3.0 

func (t *Tensor) MaxPool1D(kernel int, opt ...PoolOpt) *Tensor

MaxPool1D returns a new tensor with the result of applying a 1D max pooling operation on the input tensor. kernel: kernel size stride: stride, default 1 padding: padding, default 0 dilation: dilation, default 1 ceil: ceil, default false

func (*Tensor) MaxPool2D added in v1.3.0 

func (t *Tensor) MaxPool2D(kernel int, opt ...PoolOpt) *Tensor

MaxPool2D returns a new tensor with the result of applying a 2D max pooling operation on the input tensor. kernel: kernel size stride: stride, default 1 padding: padding, default 0 dilation: dilation, default 1 ceil: ceil, default false

func (*Tensor) MaxPool3D added in v1.3.0 

func (t *Tensor) MaxPool3D(kernel int, opt ...PoolOpt) *Tensor

MaxPool3D returns a new tensor with the result of applying a 3D max pooling operation on the input tensor. kernel: kernel size stride: stride, default 1 padding: padding, default 0 dilation: dilation, default 1 ceil: ceil, default false

func (*Tensor) Mean 

func (t *Tensor) Mean(dim int64, keepdim bool) *Tensor

func (*Tensor) Min 

func (t *Tensor) Min(dim int64, keepdim bool) *Tensor

func (*Tensor) Mul 

func (t *Tensor) Mul(t2 *Tensor) *Tensor

func (*Tensor) NArrow 

func (t *Tensor) NArrow(dim, start, length int64) *Tensor

func (*Tensor) Neg 

func (t *Tensor) Neg() *Tensor

func (*Tensor) Permute added in v1.0.3 

func (t *Tensor) Permute(dims ...int64) *Tensor

func (*Tensor) Pow 

func (t *Tensor) Pow(n float64) *Tensor

func (*Tensor) Print added in v1.3.1 

func (t *Tensor) Print()

func (*Tensor) RSqrt added in v1.5.7 

func (t *Tensor) RSqrt() *Tensor

func (*Tensor) Relu 

func (t *Tensor) Relu() *Tensor

func (*Tensor) Reshape 

func (t *Tensor) Reshape(shape ...int64) *Tensor

func (*Tensor) ScalarType 

func (t *Tensor) ScalarType() consts.ScalarType

func (*Tensor) SetRequiresGrad 

func (t *Tensor) SetRequiresGrad(b bool)

func (*Tensor) Shapes 

func (t *Tensor) Shapes() []int64

func (*Tensor) Sigmoid 

func (t *Tensor) Sigmoid() *Tensor

func (*Tensor) Silu added in v1.6.2 

func (t *Tensor) Silu() *Tensor

func (*Tensor) Softmax 

func (t *Tensor) Softmax(dim int64) *Tensor

func (*Tensor) Softmax1 added in v1.3.10 

func (t *Tensor) Softmax1(dim int64) *Tensor

func (*Tensor) Sqrt 

func (t *Tensor) Sqrt() *Tensor

func (*Tensor) Squeeze added in v1.3.3 

func (t *Tensor) Squeeze(dim int64) *Tensor

func (*Tensor) Sub 

func (t *Tensor) Sub(t2 *Tensor) *Tensor

func (*Tensor) Sum 

func (t *Tensor) Sum(dim int64, keepdim bool) *Tensor

func (*Tensor) Tanh 

func (t *Tensor) Tanh() *Tensor

func (*Tensor) Tensor 

func (t *Tensor) Tensor() torch.Tensor

func (*Tensor) ToDevice added in v1.2.0 

func (t *Tensor) ToDevice(device consts.DeviceType) *Tensor

func (*Tensor) ToScalarType added in v1.5.6 

func (t *Tensor) ToScalarType(scalarType consts.ScalarType) *Tensor

func (*Tensor) Transpose 

func (t *Tensor) Transpose(dim1, dim2 int64) *Tensor

func (*Tensor) Uint8Value 

func (t *Tensor) Uint8Value() []uint8

func (*Tensor) Unsqueeze added in v1.3.2 

func (t *Tensor) Unsqueeze(dim int64) *Tensor

func (*Tensor) Var added in v1.0.1 

func (t *Tensor) Var(dim int64, unbiased, keepdim bool) *Tensor

func (*Tensor) View added in v1.0.3 

func (t *Tensor) View(shapes ...int64) *Tensor

func (*Tensor) ViewAsComplex added in v1.6.3 

func (t *Tensor) ViewAsComplex() *Tensor

func (*Tensor) ViewAsReal added in v1.6.3 

func (t *Tensor) ViewAsReal() *Tensor

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.