libtch

package

v0.9.1 Latest Latest Go to latest Published: Oct 30, 2023 License: Apache-2.0 Imports: 7 Imported by: 2

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/sugarme/gotch

Links

Open Source Insights

README ¶

NOTES ON WRITING WRAPPER FUNCTIONS

Function Input Arguments

`tensor` -> `t *C_tensor`

void at_print(tensor);

func AtPrint(t *C_tensor) {
	c_tensor := (C.tensor)((*t).private)
	C.at_print(c_tensor)
}

C pointer e.g `int64_t *` -> `ptr unsafe.Pointer`

In function body, cPtr := (*C.long)(ptr)

void at_shape(tensor, int64_t *);

func AtShape(t *C_tensor, ptr unsafe.Pointer) {
	c_tensor := (C.tensor)((*t).private)
	c_ptr := (*C.long)(ptr)
	C.at_shape(c_tensor, c_ptr)
}

C types e.g `size_t ndims` -> equivalent Go types `ndims uint`

In function body, c_ndims := *(*C.size_t)(unsafe.Pointer(&ndims))

tensor at_tensor_of_data(void *vs, int64_t *dims, size_t ndims, size_t element_size_in_bytes, int type);

func AtTensorOfData(vs unsafe.Pointer, dims []int64, ndims uint, elt_size_in_bytes uint, kind int) *C_tensor {

    // 1. Unsafe pointer
	c_dims := (*C.int64_t)(unsafe.Pointer(&dims[0]))
	c_ndims := *(*C.size_t)(unsafe.Pointer(&ndims))
	c_elt_size_in_bytes := *(*C.size_t)(unsafe.Pointer(&elt_size_in_bytes))
	c_kind := *(*C.int)(unsafe.Pointer(&kind))

    // 2. Call C function
	t := C.at_tensor_of_data(vs, c_dims, c_ndims, c_elt_size_in_bytes, c_kind)

    // 3. Form return value
	return &C_tensor{private: unsafe.Pointer(t)}
}

Function Return

`void *CFUNC(...)`

void *at_data_ptr(tensor);

func AtDataPtr(t *C_tensor) unsafe.Pointer {
	c_tensor := (C.tensor)((*t).private)
	return C.at_data_ptr(c_tensor)
}

`tensor` -> `*C_tensor`

then in the return of function body

    // Call C function
    t := C.FUNCTION_TO_CALL(...)
    // Return
	return &C_tensor{private: unsafe.Pointer(t)}

`tensor *CFUNC(...)`

The pattern of return tensor pointer: tensor *atg_FUNCTION_NAME(). The returning tensor pointer actually is the FIRST element of a vector of C tensor pointers. Next pointer will be calculated from the first. In C land, verifying a valid pointer is to check whether it points to NULL.


tensor *atg_split(tensor self, int64_t split_size, int64_t dim);


// Wrapper
func AtgSplit(self Ctensor, splitSize int64, dim int64) *Ctensor {

	csplitSize := *(*C.int64_t)(unsafe.Pointer(&splitSize))
	cdim := *(*C.int64_t)(unsafe.Pointer(&dim))

	return C.atg_split(self, csplitSize, cdim)
}

// API

// Split splits tensor into chunks
//
// Parameters:
//  - splitSize – size of a single chunk or list of sizes for each chunk
//  - dim – dimension along which to split the tensor.
// Ref. https://pytorch.org/docs/stable/generated/torch.split.html
func (ts Tensor) Split(splitSize, dim int64) (retVal []Tensor, err error) {

	ctensorsPtr := lib.AtgSplit(ts.ctensor, splitSize, dim)
	if err = TorchErr(); err != nil {
		return retVal, err
	}

	// NOTE: ctensorsPtr is a c-pointer to a vector of tensors. The first
	// C tensor is the `ctensorsPtr` value. The next pointer will be
	// calculated from there. The vector of tensors will end if the calculated
	// pointer value is `null`.
	currentPtr := ctensorsPtr
	retVal = append(retVal, Tensor{ctensor: *currentPtr})
	for {
		// calculate the next pointer value
		nextPtr := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(currentPtr)) + unsafe.Sizeof(currentPtr)))
		if *nextPtr == nil {
			break
		}

		retVal = append(retVal, Tensor{ctensor: *nextPtr})
		currentPtr = nextPtr
	}

	return retVal, nil
}

C types e.g. `C_ulong` -> Go equivalent types `uint64`

then in the return of function body


	c_result := C.FUNCTION_CALL(...)
	return *(*uint64)(unsafe.Pointer(&c_result))

C type pointers e.g. `char FUNCTION()` --> `C.char`

then just return the C function call.

char *get_and_reset_last_err(); // thread-local

func GetAndResetLastErr() *C.char{
   return C.get_and_reset_last_err()
}

Multiple Tensors Created In C Land Memory

When there are multiple Ctensor created in C land memory. A first Ctensor pointer will be created and given to the C function. It will create consecutive Ctensor(s) based on this pointer. The next pointer(s) can be calulated based on this pointer and its size.

Example: lstm function

C function

    void atg_lstm(tensor *, tensor input, tensor *hx_data, int hx_len, tensor *params_data, int params_len, int has_biases, int64_t num_layers, double dropout, int train, int bidirectional, int batch_first);

Go wrapper function

    func AtgLstm(ptr *Ctensor, input Ctensor, hxData []Ctensor, hxLen int, paramsData []Ctensor, paramsLen int, hasBiases int, numLayers int64, dropout float64, train int, bidirectional int, batchFirst int) {

        chxDataPtr := (*Ctensor)(unsafe.Pointer(&hxData[0]))
        chxLen := *(*C.int)(unsafe.Pointer(&hxLen))
        cparamsDataPtr := (*Ctensor)(unsafe.Pointer(&paramsData[0]))
        cparamsLen := *(*C.int)(unsafe.Pointer(&paramsLen))
        chasBiases := *(*C.int)(unsafe.Pointer(&hasBiases))
        cnumLayers := *(*C.int64_t)(unsafe.Pointer(&numLayers))
        cdropout := *(*C.double)(unsafe.Pointer(&dropout))
        ctrain := *(*C.int)(unsafe.Pointer(&train))
        cbidirectional := *(*C.int)(unsafe.Pointer(&bidirectional))
        cbatchFirst := *(*C.int)(unsafe.Pointer(&batchFirst))

        C.atg_lstm(ptr, input, chxDataPtr, chxLen, cparamsDataPtr, cparamsLen, chasBiases, cnumLayers, cdropout, ctrain, cbidirectional, cbatchFirst)
    }

Go API function

func (ts Tensor) LSTM(hxData []Tensor, paramsData []Tensor, hasBiases bool, numLayers int64, dropout float64, train bool, bidirectional bool, batchFirst bool) (output, h, c Tensor, err error) {

    // NOTE: `atg_lstm` will create 3 consecutive Ctensors in memory of C land. The first
    // Ctensor will have address given by `ctensorPtr1` here.
    // The next pointers can be calculated based on `ctensorPtr1`
    ctensorPtr1 := (*lib.Ctensor)(unsafe.Pointer(C.malloc(0)))
    ctensorPtr2 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr1)) + unsafe.Sizeof(ctensorPtr1)))
    ctensorPtr3 := (*lib.Ctensor)(unsafe.Pointer(uintptr(unsafe.Pointer(ctensorPtr2)) + unsafe.Sizeof(ctensorPtr1)))

    var chxData []lib.Ctensor
    for _, t := range hxData {
        chxData = append(chxData, t.ctensor)
    }

    var cparamsData []lib.Ctensor
    for _, t := range paramsData {
        cparamsData = append(cparamsData, t.ctensor)
    }

    chasBiases := 0
    if hasBiases {
        chasBiases = 1
    }
    ctrain := 0
    if train {
        ctrain = 1
    }
    cbidirectional := 0
    if bidirectional {
        cbidirectional = 1
    }
    cbatchFirst := 0
    if batchFirst {
        cbatchFirst = 1
    }

    lib.AtgLstm(ctensorPtr1, ts.ctensor, chxData, len(hxData), cparamsData, len(paramsData), chasBiases, numLayers, dropout, ctrain, cbidirectional, cbatchFirst)
    err = TorchErr()
    if err != nil {
        return output, h, c, err
    }

    return Tensor{ctensor: *ctensorPtr1}, Tensor{ctensor: *ctensorPtr2}, Tensor{ctensor: *ctensorPtr3}, nil

}

Documentation ¶

Overview ¶

NOTE: functions in this file would be automatically generated and named as `c-generated.go`

Index ¶

Variables
func AtBackward(ts Ctensor, keepGraph int, createGraph int)
func AtCopyData(ts Ctensor, vs unsafe.Pointer, numel uint, element_size_in_bytes uint)
func AtCopy_(dst Ctensor, src Ctensor)
func AtDataPtr(t Ctensor) unsafe.Pointer
func AtDefined(ts Ctensor) bool
func AtDevice(ts Ctensor) int
func AtDim(t Ctensor) uint64
func AtDoubleValueAtIndexes(ts Ctensor, indexes unsafe.Pointer, indexesLen int) float64
func AtFree(ts Ctensor)
func AtGradSetEnabled(b int) int
func AtInt64ValueAtIndexes(ts Ctensor, indexes unsafe.Pointer, indexesLen int) int64
func AtIsContiguous(ts Ctensor) bool
func AtIsMkldnn(ts Ctensor) bool
func AtIsSparse(ts Ctensor) bool
func AtLoadCallback(filename string, dataPtr unsafe.Pointer)
func AtLoadCallbackWithDevice(filename string, dataPtr unsafe.Pointer, device int32)
func AtLoadMulti(tensors []Ctensor, tensor_names []string, ntensors int, filename string)
func AtPrint(t Ctensor)
func AtRequiresGrad(ts Ctensor) bool
func AtRunBackward(tensorsPtr *Ctensor, ntensors int, inputsPtr *Ctensor, ninputs int, ...)
func AtSave(ts Ctensor, path string)
func AtSaveImage(ts Ctensor, path string)
func AtSaveMulti(tensors []Ctensor, tensor_names []string, ntensors int, filename string)
func AtSaveMultiNew(tensors []Ctensor, names []string, filename string)
func AtScalarType(t Ctensor) int32
func AtShape(t Ctensor, ptr unsafe.Pointer)
func AtStride(t Ctensor, ptr unsafe.Pointer)
func AtToString(ts Ctensor, lineSize int64) string
func AtcCudaDeviceCount() int32
func AtcCudaIsAvailable() bool
func AtcCudnnIsAvailable() bool
func AtcGetDevice() int
func AtcSetBenchmarkCudnn(b int)
func AtcSetDevice(deviceIndex int) int
func AtcSynchronize(deviceIndex int64)
func Atc_cuda_device_count() int
func AtgAbs(ptr *Ctensor, self Ctensor)
func AtgAbsOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAbs_(ptr *Ctensor, self Ctensor)
func AtgAbsolute(ptr *Ctensor, self Ctensor)
func AtgAbsoluteOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAbsolute_(ptr *Ctensor, self Ctensor)
func AtgAcos(ptr *Ctensor, self Ctensor)
func AtgAcosOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAcos_(ptr *Ctensor, self Ctensor)
func AtgAcosh(ptr *Ctensor, self Ctensor)
func AtgAcoshOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAcosh_(ptr *Ctensor, self Ctensor)
func AtgAdaptiveAvgPool1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)
func AtgAdaptiveAvgPool2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)
func AtgAdaptiveAvgPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgAdaptiveAvgPool3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)
func AtgAdaptiveAvgPool3dBackward(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor)
func AtgAdaptiveAvgPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgAdaptiveMaxPool1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)
func AtgAdaptiveMaxPool2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)
func AtgAdaptiveMaxPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, indices Ctensor)
func AtgAdaptiveMaxPool2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgAdaptiveMaxPool2dOut(ptr *Ctensor, out Ctensor, indices Ctensor, self Ctensor, ...)
func AtgAdaptiveMaxPool3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)
func AtgAdaptiveMaxPool3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, indices Ctensor)
func AtgAdaptiveMaxPool3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgAdaptiveMaxPool3dOut(ptr *Ctensor, out Ctensor, indices Ctensor, self Ctensor, ...)
func AtgAdd(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgAddOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgAddScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgAddScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgAddScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgAdd_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgAddbmm(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)
func AtgAddbmmOut(ptr *Ctensor, out Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)
func AtgAddbmm_(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)
func AtgAddcdiv(ptr *Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)
func AtgAddcdivOut(ptr *Ctensor, out Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)
func AtgAddcdiv_(ptr *Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)
func AtgAddcmul(ptr *Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)
func AtgAddcmulOut(ptr *Ctensor, out Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)
func AtgAddcmul_(ptr *Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)
func AtgAddmm(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)
func AtgAddmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)
func AtgAddmm_(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)
func AtgAddmv(ptr *Ctensor, self Ctensor, mat Ctensor, vec Ctensor)
func AtgAddmvOut(ptr *Ctensor, out Ctensor, self Ctensor, mat Ctensor, vec Ctensor)
func AtgAddmv_(ptr *Ctensor, self Ctensor, mat Ctensor, vec Ctensor)
func AtgAddr(ptr *Ctensor, self Ctensor, vec1 Ctensor, vec2 Ctensor)
func AtgAddrOut(ptr *Ctensor, out Ctensor, self Ctensor, vec1 Ctensor, vec2 Ctensor)
func AtgAddr_(ptr *Ctensor, self Ctensor, vec1 Ctensor, vec2 Ctensor)
func AtgAdjoint(ptr *Ctensor, self Ctensor)
func AtgAffineGridGenerator(ptr *Ctensor, theta Ctensor, sizeData []int64, sizeLen int, alignCorners int32)
func AtgAffineGridGeneratorBackward(ptr *Ctensor, grad Ctensor, sizeData []int64, sizeLen int, alignCorners int32)
func AtgAffineGridGeneratorOut(ptr *Ctensor, out Ctensor, theta Ctensor, sizeData []int64, sizeLen int, ...)
func AtgAlias(ptr *Ctensor, self Ctensor)
func AtgAliasCopy(ptr *Ctensor, self Ctensor)
func AtgAliasCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAlignAs(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgAll(ptr *Ctensor, self Ctensor)
func AtgAllAllOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAllDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)
func AtgAllOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, keepdim int32)
func AtgAllclose(self Ctensor, other Ctensor, rtol float64, atol float64, equalNan int32) bool
func AtgAlphaDropout(ptr *Ctensor, input Ctensor, p float64, train int32)
func AtgAlphaDropout_(ptr *Ctensor, self Ctensor, p float64, train int32)
func AtgAmax(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)
func AtgAmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgAmin(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)
func AtgAminOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgAminmax(ptr *Ctensor, self Ctensor, dimVal int64, dimNull int, keepdim int32)
func AtgAminmaxOut(ptr *Ctensor, min Ctensor, max Ctensor, self Ctensor, dimVal int64, ...)
func AtgAngle(ptr *Ctensor, self Ctensor)
func AtgAngleOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAny(ptr *Ctensor, self Ctensor)
func AtgAnyAllOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAnyDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)
func AtgAnyOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, keepdim int32)
func AtgArange(ptr *Ctensor, end Cscalar, optionsKind int32, optionsDevice int32)
func AtgArangeStart(ptr *Ctensor, start Cscalar, end Cscalar, optionsKind int32, ...)
func AtgArangeStartStep(ptr *Ctensor, start Cscalar, end Cscalar, step Cscalar, optionsKind int32, ...)
func AtgArccos(ptr *Ctensor, self Ctensor)
func AtgArccosOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgArccos_(ptr *Ctensor, self Ctensor)
func AtgArccosh(ptr *Ctensor, self Ctensor)
func AtgArccoshOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgArccosh_(ptr *Ctensor, self Ctensor)
func AtgArcsin(ptr *Ctensor, self Ctensor)
func AtgArcsinOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgArcsin_(ptr *Ctensor, self Ctensor)
func AtgArcsinh(ptr *Ctensor, self Ctensor)
func AtgArcsinhOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgArcsinh_(ptr *Ctensor, self Ctensor)
func AtgArctan(ptr *Ctensor, self Ctensor)
func AtgArctan2(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgArctan2Out(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgArctan2_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgArctanOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgArctan_(ptr *Ctensor, self Ctensor)
func AtgArctanh(ptr *Ctensor, self Ctensor)
func AtgArctanhOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgArctanh_(ptr *Ctensor, self Ctensor)
func AtgArgmax(ptr *Ctensor, self Ctensor, dimVal int64, dimNull int, keepdim int32)
func AtgArgmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dimVal int64, dimNull int, ...)
func AtgArgmin(ptr *Ctensor, self Ctensor, dimVal int64, dimNull int, keepdim int32)
func AtgArgminOut(ptr *Ctensor, out Ctensor, self Ctensor, dimVal int64, dimNull int, ...)
func AtgArgsort(ptr *Ctensor, self Ctensor, dim int64, descending int32)
func AtgArgsortStable(ptr *Ctensor, self Ctensor, stable int32, dim int64, descending int32)
func AtgArgsortStableOut(ptr *Ctensor, out Ctensor, self Ctensor, stable int32, dim int64, ...)
func AtgArgwhere(ptr *Ctensor, self Ctensor)
func AtgAsStrided(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, ...)
func AtgAsStridedCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, ...)
func AtgAsStridedCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, ...)
func AtgAsStridedScatter(ptr *Ctensor, self Ctensor, src Ctensor, sizeData []int64, sizeLen int, ...)
func AtgAsStridedScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, sizeData []int64, ...)
func AtgAsStrided_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, ...)
func AtgAsin(ptr *Ctensor, self Ctensor)
func AtgAsinOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAsin_(ptr *Ctensor, self Ctensor)
func AtgAsinh(ptr *Ctensor, self Ctensor)
func AtgAsinhOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAsinh_(ptr *Ctensor, self Ctensor)
func AtgAtan(ptr *Ctensor, self Ctensor)
func AtgAtan2(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgAtan2Out(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgAtan2_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgAtanOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAtan_(ptr *Ctensor, self Ctensor)
func AtgAtanh(ptr *Ctensor, self Ctensor)
func AtgAtanhOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgAtanh_(ptr *Ctensor, self Ctensor)
func AtgAtleast1d(ptr *Ctensor, self Ctensor)
func AtgAtleast2d(ptr *Ctensor, self Ctensor)
func AtgAtleast3d(ptr *Ctensor, self Ctensor)
func AtgAvgPool1d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgAvgPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgAvgPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgAvgPool2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgAvgPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgAvgPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgAvgPool3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgAvgPool3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgAvgPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgBaddbmm(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor, beta Cscalar, ...)
func AtgBaddbmmOut(ptr *Ctensor, out Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)
func AtgBaddbmm_(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)
func AtgBartlettWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)
func AtgBartlettWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)
func AtgBartlettWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, ...)
func AtgBartlettWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)
func AtgBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, ...)
func AtgBatchNormBackwardElemt(ptr *Ctensor, gradOut Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, ...)
func AtgBatchNormBackwardElemtOut(ptr *Ctensor, out Ctensor, gradOut Ctensor, input Ctensor, mean Ctensor, ...)
func AtgBatchNormBackwardReduce(ptr *Ctensor, gradOut Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, ...)
func AtgBatchNormBackwardReduceOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, ...)
func AtgBatchNormElemt(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, mean Ctensor, ...)
func AtgBatchNormElemtOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgBatchNormGatherStats(ptr *Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, runningMean Ctensor, ...)
func AtgBatchNormGatherStatsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, mean Ctensor, ...)
func AtgBatchNormGatherStatsWithCounts(ptr *Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, runningMean Ctensor, ...)
func AtgBatchNormGatherStatsWithCountsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, mean Ctensor, ...)
func AtgBatchNormStats(ptr *Ctensor, input Ctensor, eps float64)
func AtgBatchNormStatsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, eps float64)
func AtgBatchNormUpdateStats(ptr *Ctensor, input Ctensor, runningMean Ctensor, runningVar Ctensor, ...)
func AtgBatchNormUpdateStatsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, runningMean Ctensor, ...)
func AtgBernoulli(ptr *Ctensor, self Ctensor)
func AtgBernoulliFloat_(ptr *Ctensor, self Ctensor, p float64)
func AtgBernoulliP(ptr *Ctensor, self Ctensor, p float64)
func AtgBernoulliTensor(ptr *Ctensor, self Ctensor, p Ctensor)
func AtgBernoulli_(ptr *Ctensor, self Ctensor, p Ctensor)
func AtgBilinear(ptr *Ctensor, input1 Ctensor, input2 Ctensor, weight Ctensor, bias Ctensor)
func AtgBinaryCrossEntropy(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64)
func AtgBinaryCrossEntropyBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, weight Ctensor, ...)
func AtgBinaryCrossEntropyBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgBinaryCrossEntropyOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, weight Ctensor, ...)
func AtgBinaryCrossEntropyWithLogits(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, posWeight Ctensor, ...)
func AtgBinaryCrossEntropyWithLogitsOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, weight Ctensor, ...)
func AtgBincount(ptr *Ctensor, self Ctensor, weights Ctensor, minlength int64)
func AtgBincountOut(ptr *Ctensor, out Ctensor, self Ctensor, weights Ctensor, minlength int64)
func AtgBinomial(ptr *Ctensor, count Ctensor, prob Ctensor)
func AtgBinomialOut(ptr *Ctensor, out Ctensor, count Ctensor, prob Ctensor)
func AtgBitwiseAnd(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseAndScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseAndScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseAndScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseAndTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseAndTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseAndTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseAnd_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseLeftShift(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseLeftShiftScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseLeftShiftScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseLeftShiftTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseLeftShiftTensorScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseLeftShiftTensorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseLeftShiftTensorScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseLeftShift_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseNot(ptr *Ctensor, self Ctensor)
func AtgBitwiseNotOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgBitwiseNot_(ptr *Ctensor, self Ctensor)
func AtgBitwiseOr(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseOrScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseOrScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseOrScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseOrTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseOrTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseOrTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseOr_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseRightShift(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseRightShiftScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseRightShiftScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseRightShiftTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseRightShiftTensorScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseRightShiftTensorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseRightShiftTensorScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseRightShift_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseXor(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseXorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgBitwiseXorScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseXorScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgBitwiseXorTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseXorTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseXorTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgBitwiseXor_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgBlackmanWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)
func AtgBlackmanWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)
func AtgBlackmanWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, ...)
func AtgBlackmanWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)
func AtgBlockDiag(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgBlockDiagOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgBmm(ptr *Ctensor, self Ctensor, mat2 Ctensor)
func AtgBmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor)
func AtgBroadcastTo(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgBucketize(ptr *Ctensor, self Ctensor, boundaries Ctensor, outInt32 int32, right int32)
func AtgBucketizeScalar(ptr *Ctensor, selfScalar Cscalar, boundaries Ctensor, outInt32 int32, ...)
func AtgBucketizeScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, boundaries Ctensor, ...)
func AtgBucketizeTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, boundaries Ctensor, outInt32 int32, ...)
func AtgCanCast(from int32, to int32) bool
func AtgCartesianProd(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgCat(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func AtgCatOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func AtgCauchy(ptr *Ctensor, self Ctensor, median float64, sigma float64)
func AtgCauchyOut(ptr *Ctensor, out Ctensor, self Ctensor, median float64, sigma float64)
func AtgCauchy_(ptr *Ctensor, self Ctensor, median float64, sigma float64)
func AtgCcolIndices(ptr *Ctensor, self Ctensor)
func AtgCcolIndicesCopy(ptr *Ctensor, self Ctensor)
func AtgCcolIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgCdist(ptr *Ctensor, x1 Ctensor, x2 Ctensor, p float64, computeModeVal int64, ...)
func AtgCeil(ptr *Ctensor, self Ctensor)
func AtgCeilOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgCeil_(ptr *Ctensor, self Ctensor)
func AtgCelu(ptr *Ctensor, self Ctensor)
func AtgCeluOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgCelu_(ptr *Ctensor, self Ctensor)
func AtgChainMatmul(ptr *Ctensor, matricesData []Ctensor, matricesLen int)
func AtgChainMatmulOut(ptr *Ctensor, out Ctensor, matricesData []Ctensor, matricesLen int)
func AtgChalf(ptr *Ctensor, self Ctensor)
func AtgChannelShuffle(ptr *Ctensor, self Ctensor, groups int64)
func AtgChannelShuffleOut(ptr *Ctensor, out Ctensor, self Ctensor, groups int64)
func AtgCholesky(ptr *Ctensor, self Ctensor, upper int32)
func AtgCholeskyInverse(ptr *Ctensor, self Ctensor, upper int32)
func AtgCholeskyInverseOut(ptr *Ctensor, out Ctensor, self Ctensor, upper int32)
func AtgCholeskyOut(ptr *Ctensor, out Ctensor, self Ctensor, upper int32)
func AtgCholeskySolve(ptr *Ctensor, self Ctensor, input2 Ctensor, upper int32)
func AtgCholeskySolveOut(ptr *Ctensor, out Ctensor, self Ctensor, input2 Ctensor, upper int32)
func AtgChooseQparamsOptimized(ptr *Ctensor, input Ctensor, numel int64, nBins int64, ratio float64, ...)
func AtgClamp(ptr *Ctensor, self Ctensor, min Cscalar, max Cscalar)
func AtgClampMax(ptr *Ctensor, self Ctensor, max Cscalar)
func AtgClampMaxOut(ptr *Ctensor, out Ctensor, self Ctensor, max Cscalar)
func AtgClampMaxTensor(ptr *Ctensor, self Ctensor, max Ctensor)
func AtgClampMaxTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, max Ctensor)
func AtgClampMaxTensor_(ptr *Ctensor, self Ctensor, max Ctensor)
func AtgClampMax_(ptr *Ctensor, self Ctensor, max Cscalar)
func AtgClampMin(ptr *Ctensor, self Ctensor, min Cscalar)
func AtgClampMinOut(ptr *Ctensor, out Ctensor, self Ctensor, min Cscalar)
func AtgClampMinTensor(ptr *Ctensor, self Ctensor, min Ctensor)
func AtgClampMinTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, min Ctensor)
func AtgClampMinTensor_(ptr *Ctensor, self Ctensor, min Ctensor)
func AtgClampMin_(ptr *Ctensor, self Ctensor, min Cscalar)
func AtgClampOut(ptr *Ctensor, out Ctensor, self Ctensor, min Cscalar, max Cscalar)
func AtgClampTensor(ptr *Ctensor, self Ctensor, min Ctensor, max Ctensor)
func AtgClampTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, min Ctensor, max Ctensor)
func AtgClampTensor_(ptr *Ctensor, self Ctensor, min Ctensor, max Ctensor)
func AtgClamp_(ptr *Ctensor, self Ctensor, min Cscalar, max Cscalar)
func AtgClip(ptr *Ctensor, self Ctensor, min Cscalar, max Cscalar)
func AtgClipOut(ptr *Ctensor, out Ctensor, self Ctensor, min Cscalar, max Cscalar)
func AtgClipTensor(ptr *Ctensor, self Ctensor, min Ctensor, max Ctensor)
func AtgClipTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, min Ctensor, max Ctensor)
func AtgClipTensor_(ptr *Ctensor, self Ctensor, min Ctensor, max Ctensor)
func AtgClip_(ptr *Ctensor, self Ctensor, min Cscalar, max Cscalar)
func AtgClone(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgCoalesce(ptr *Ctensor, self Ctensor)
func AtgCol2im(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgCol2imOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgColIndices(ptr *Ctensor, self Ctensor)
func AtgColIndicesCopy(ptr *Ctensor, self Ctensor)
func AtgColIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgColumnStack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgColumnStackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgCombinations(ptr *Ctensor, self Ctensor, r int64, withReplacement int32)
func AtgComplex(ptr *Ctensor, real Ctensor, imag Ctensor)
func AtgComplexOut(ptr *Ctensor, out Ctensor, real Ctensor, imag Ctensor)
func AtgConcat(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func AtgConcatOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func AtgConcatenate(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func AtgConcatenateOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func AtgConj(ptr *Ctensor, self Ctensor)
func AtgConjPhysical(ptr *Ctensor, self Ctensor)
func AtgConjPhysicalOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgConjPhysical_(ptr *Ctensor, self Ctensor)
func AtgConstantPadNd(ptr *Ctensor, self Ctensor, padData []int64, padLen int)
func AtgConstantPadNdOut(ptr *Ctensor, out Ctensor, self Ctensor, padData []int64, padLen int)
func AtgContiguous(ptr *Ctensor, self Ctensor)
func AtgConv1d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConv1dPadding(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConv2d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConv2dPadding(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConv3d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConv3dPadding(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConvDepthwise3d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, ...)
func AtgConvDepthwise3dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, ...)
func AtgConvTbc(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, pad int64)
func AtgConvTbcBackward(ptr *Ctensor, self Ctensor, input Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgConvTbcOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgConvTranspose1d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConvTranspose2d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConvTranspose3d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConvolution(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConvolutionOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgConvolutionOverrideable(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgConvolutionOverrideableOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgCopy(ptr *Ctensor, self Ctensor, src Ctensor, nonBlocking int32)
func AtgCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, nonBlocking int32)
func AtgCopySparseToSparse(ptr *Ctensor, self Ctensor, src Ctensor, nonBlocking int32)
func AtgCopySparseToSparseOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, nonBlocking int32)
func AtgCopySparseToSparse_(ptr *Ctensor, self Ctensor, src Ctensor, nonBlocking int32)
func AtgCopysign(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgCopysignOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgCopysignScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgCopysignScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgCopysignScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgCopysign_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgCorrcoef(ptr *Ctensor, self Ctensor)
func AtgCos(ptr *Ctensor, self Ctensor)
func AtgCosOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgCos_(ptr *Ctensor, self Ctensor)
func AtgCosh(ptr *Ctensor, self Ctensor)
func AtgCoshOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgCosh_(ptr *Ctensor, self Ctensor)
func AtgCosineEmbeddingLoss(ptr *Ctensor, input1 Ctensor, input2 Ctensor, target Ctensor, margin float64, ...)
func AtgCosineSimilarity(ptr *Ctensor, x1 Ctensor, x2 Ctensor, dim int64, eps float64)
func AtgCountNonzero(ptr *Ctensor, self Ctensor, dimVal int64, dimNull int)
func AtgCountNonzeroDimIntlist(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)
func AtgCountNonzeroDimIntlistOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int)
func AtgCountNonzeroOut(ptr *Ctensor, out Ctensor, self Ctensor, dimVal int64, dimNull int)
func AtgCov(ptr *Ctensor, self Ctensor, correction int64, fweights Ctensor, ...)
func AtgCross(ptr *Ctensor, self Ctensor, other Ctensor, dimVal int64, dimNull int)
func AtgCrossEntropyLoss(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ...)
func AtgCrossOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, dimVal int64, ...)
func AtgCrowIndices(ptr *Ctensor, self Ctensor)
func AtgCrowIndicesCopy(ptr *Ctensor, self Ctensor)
func AtgCrowIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgCtcLoss(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, ...)
func AtgCtcLossTensor(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengths Ctensor, ...)
func AtgCudnnAffineGridGenerator(ptr *Ctensor, theta Ctensor, n int64, c int64, h int64, w int64)
func AtgCudnnAffineGridGeneratorBackward(ptr *Ctensor, grad Ctensor, n int64, c int64, h int64, w int64)
func AtgCudnnAffineGridGeneratorBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, n int64, c int64, h int64, w int64)
func AtgCudnnAffineGridGeneratorOut(ptr *Ctensor, out Ctensor, theta Ctensor, n int64, c int64, h int64, w int64)
func AtgCudnnBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, ...)
func AtgCudnnBatchNormBackward(ptr *Ctensor, input Ctensor, gradOutput Ctensor, weight Ctensor, ...)
func AtgCudnnBatchNormBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, ...)
func AtgCudnnBatchNormOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, ...)
func AtgCudnnConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, paddingData []int64, ...)
func AtgCudnnConvolutionAddRelu(ptr *Ctensor, self Ctensor, weight Ctensor, z Ctensor, alpha Cscalar, ...)
func AtgCudnnConvolutionAddReluOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, z Ctensor, ...)
func AtgCudnnConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, paddingData []int64, ...)
func AtgCudnnConvolutionRelu(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgCudnnConvolutionReluOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgCudnnConvolutionTranspose(ptr *Ctensor, self Ctensor, weight Ctensor, paddingData []int64, ...)
func AtgCudnnConvolutionTransposeOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, paddingData []int64, ...)
func AtgCudnnGridSampler(ptr *Ctensor, self Ctensor, grid Ctensor)
func AtgCudnnGridSamplerBackward(ptr *Ctensor, self Ctensor, grid Ctensor, gradOutput Ctensor)
func AtgCudnnGridSamplerBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, grid Ctensor, ...)
func AtgCudnnGridSamplerOut(ptr *Ctensor, out Ctensor, self Ctensor, grid Ctensor)
func AtgCudnnIsAcceptable(self Ctensor) bool
func AtgCummax(ptr *Ctensor, self Ctensor, dim int64)
func AtgCummaxOut(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64)
func AtgCummaxminBackward(ptr *Ctensor, grad Ctensor, input Ctensor, indices Ctensor, dim int64)
func AtgCummin(ptr *Ctensor, self Ctensor, dim int64)
func AtgCumminOut(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64)
func AtgCumprod(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func AtgCumprodBackward(ptr *Ctensor, grad Ctensor, input Ctensor, dim int64, output Ctensor)
func AtgCumprodOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, dtype int32)
func AtgCumprod_(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func AtgCumsum(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func AtgCumsumOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, dtype int32)
func AtgCumsum_(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func AtgCumulativeTrapezoid(ptr *Ctensor, y Ctensor, dim int64)
func AtgCumulativeTrapezoidX(ptr *Ctensor, y Ctensor, x Ctensor, dim int64)
func AtgData(ptr *Ctensor, self Ctensor)
func AtgDeg2rad(ptr *Ctensor, self Ctensor)
func AtgDeg2radOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgDeg2rad_(ptr *Ctensor, self Ctensor)
func AtgDenseDim(self Ctensor) int64
func AtgDequantize(ptr *Ctensor, self Ctensor)
func AtgDet(ptr *Ctensor, self Ctensor)
func AtgDetach(ptr *Ctensor, self Ctensor)
func AtgDetachCopy(ptr *Ctensor, self Ctensor)
func AtgDetachCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgDetach_(ptr *Ctensor, self Ctensor)
func AtgDiag(ptr *Ctensor, self Ctensor, diagonal int64)
func AtgDiagEmbed(ptr *Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)
func AtgDiagEmbedOut(ptr *Ctensor, out Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)
func AtgDiagOut(ptr *Ctensor, out Ctensor, self Ctensor, diagonal int64)
func AtgDiagflat(ptr *Ctensor, self Ctensor, offset int64)
func AtgDiagonal(ptr *Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)
func AtgDiagonalBackward(ptr *Ctensor, gradOutput Ctensor, inputSizesData []int64, inputSizesLen int, ...)
func AtgDiagonalBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, inputSizesData []int64, ...)
func AtgDiagonalCopy(ptr *Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)
func AtgDiagonalCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)
func AtgDiagonalScatter(ptr *Ctensor, self Ctensor, src Ctensor, offset int64, dim1 int64, dim2 int64)
func AtgDiagonalScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, offset int64, dim1 int64, ...)
func AtgDiff(ptr *Ctensor, self Ctensor, n int64, dim int64, prepend Ctensor, ...)
func AtgDiffOut(ptr *Ctensor, out Ctensor, self Ctensor, n int64, dim int64, prepend Ctensor, ...)
func AtgDigamma(ptr *Ctensor, self Ctensor)
func AtgDigammaOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgDigamma_(ptr *Ctensor, self Ctensor)
func AtgDist(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgDistOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgDiv(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgDivOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgDivOutMode(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, roundingMode string)
func AtgDivScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgDivScalarMode(ptr *Ctensor, self Ctensor, other Cscalar, roundingMode string)
func AtgDivScalarModeOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar, roundingMode string)
func AtgDivScalarMode_(ptr *Ctensor, self Ctensor, other Cscalar, roundingMode string)
func AtgDivScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgDivScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgDivTensorMode(ptr *Ctensor, self Ctensor, other Ctensor, roundingMode string)
func AtgDivTensorMode_(ptr *Ctensor, self Ctensor, other Ctensor, roundingMode string)
func AtgDiv_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgDivide(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgDivideOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgDivideOutMode(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, roundingMode string)
func AtgDivideScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgDivideScalarMode(ptr *Ctensor, self Ctensor, other Cscalar, roundingMode string)
func AtgDivideScalarMode_(ptr *Ctensor, self Ctensor, other Cscalar, roundingMode string)
func AtgDivideScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgDivideTensorMode(ptr *Ctensor, self Ctensor, other Ctensor, roundingMode string)
func AtgDivideTensorMode_(ptr *Ctensor, self Ctensor, other Ctensor, roundingMode string)
func AtgDivide_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgDot(ptr *Ctensor, self Ctensor, tensor Ctensor)
func AtgDotOut(ptr *Ctensor, out Ctensor, self Ctensor, tensor Ctensor)
func AtgDropout(ptr *Ctensor, input Ctensor, p float64, train int32)
func AtgDropout_(ptr *Ctensor, self Ctensor, p float64, train int32)
func AtgDstack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgDstackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgEinsum(ptr *Ctensor, equation string, tensorsData []Ctensor, tensorsLen int, ...)
func AtgElu(ptr *Ctensor, self Ctensor)
func AtgEluBackward(ptr *Ctensor, gradOutput Ctensor, alpha Cscalar, scale Cscalar, ...)
func AtgEluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, alpha Cscalar, ...)
func AtgEluOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgElu_(ptr *Ctensor, self Ctensor)
func AtgEmbedding(ptr *Ctensor, weight Ctensor, indices Ctensor, paddingIdx int64, ...)
func AtgEmbeddingBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, numWeights int64, ...)
func AtgEmbeddingBag(ptr *Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, ...)
func AtgEmbeddingBagPaddingIdx(ptr *Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, ...)
func AtgEmbeddingDenseBackward(ptr *Ctensor, gradOutput Ctensor, indices Ctensor, numWeights int64, ...)
func AtgEmbeddingDenseBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, indices Ctensor, ...)
func AtgEmbeddingOut(ptr *Ctensor, out Ctensor, weight Ctensor, indices Ctensor, paddingIdx int64, ...)
func AtgEmbeddingRenorm(ptr *Ctensor, self Ctensor, indices Ctensor, maxNorm float64, normType float64)
func AtgEmbeddingRenormOut(ptr *Ctensor, out Ctensor, self Ctensor, indices Ctensor, maxNorm float64, ...)
func AtgEmbeddingRenorm_(ptr *Ctensor, self Ctensor, indices Ctensor, maxNorm float64, normType float64)
func AtgEmbeddingSparseBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, numWeights int64, ...)
func AtgEmpty(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgEmptyLike(ptr *Ctensor, self Ctensor)
func AtgEmptyLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgEmptyOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)
func AtgEmptyPermuted(ptr *Ctensor, sizeData []int64, sizeLen int, physicalLayoutData []int64, ...)
func AtgEmptyPermutedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, ...)
func AtgEmptyQuantized(ptr *Ctensor, sizeData []int64, sizeLen int, qtensor Ctensor, ...)
func AtgEmptyQuantizedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, qtensor Ctensor)
func AtgEmptyStrided(ptr *Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int, ...)
func AtgEmptyStridedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, strideData []int64, ...)
func AtgEq(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgEqScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgEqTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgEqTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgEqTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgEq_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgEqual(self Ctensor, other Ctensor) bool
func AtgErf(ptr *Ctensor, self Ctensor)
func AtgErfOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgErf_(ptr *Ctensor, self Ctensor)
func AtgErfc(ptr *Ctensor, self Ctensor)
func AtgErfcOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgErfc_(ptr *Ctensor, self Ctensor)
func AtgErfinv(ptr *Ctensor, self Ctensor)
func AtgErfinvOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgErfinv_(ptr *Ctensor, self Ctensor)
func AtgExp(ptr *Ctensor, self Ctensor)
func AtgExp2(ptr *Ctensor, self Ctensor)
func AtgExp2Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgExp2_(ptr *Ctensor, self Ctensor)
func AtgExpOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgExp_(ptr *Ctensor, self Ctensor)
func AtgExpand(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, implicit int32)
func AtgExpandAs(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgExpandCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, implicit int32)
func AtgExpandCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, ...)
func AtgExpm1(ptr *Ctensor, self Ctensor)
func AtgExpm1Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgExpm1_(ptr *Ctensor, self Ctensor)
func AtgExponential(ptr *Ctensor, self Ctensor, lambd float64)
func AtgExponentialOut(ptr *Ctensor, out Ctensor, self Ctensor, lambd float64)
func AtgExponential_(ptr *Ctensor, self Ctensor, lambd float64)
func AtgEye(ptr *Ctensor, n int64, optionsKind int32, optionsDevice int32)
func AtgEyeM(ptr *Ctensor, n int64, m int64, optionsKind int32, optionsDevice int32)
func AtgEyeMOut(ptr *Ctensor, out Ctensor, n int64, m int64)
func AtgEyeOut(ptr *Ctensor, out Ctensor, n int64)
func AtgFakeQuantizePerChannelAffine(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64, ...)
func AtgFakeQuantizePerChannelAffineCachemask(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64, ...)
func AtgFakeQuantizePerChannelAffineCachemaskBackward(ptr *Ctensor, grad Ctensor, mask Ctensor)
func AtgFakeQuantizePerChannelAffineCachemaskOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, scale Ctensor, ...)
func AtgFakeQuantizePerTensorAffine(ptr *Ctensor, self Ctensor, scale float64, zeroPoint int64, quantMin int64, ...)
func AtgFakeQuantizePerTensorAffineCachemask(ptr *Ctensor, self Ctensor, scale float64, zeroPoint int64, quantMin int64, ...)
func AtgFakeQuantizePerTensorAffineCachemaskBackward(ptr *Ctensor, grad Ctensor, mask Ctensor)
func AtgFakeQuantizePerTensorAffineCachemaskOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, scale float64, ...)
func AtgFakeQuantizePerTensorAffineTensorQparams(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, quantMin int64, ...)
func AtgFbgemmLinearFp16Weight(ptr *Ctensor, input Ctensor, packedWeight Ctensor, bias Ctensor)
func AtgFbgemmLinearFp16WeightFp32Activation(ptr *Ctensor, input Ctensor, packedWeight Ctensor, bias Ctensor)
func AtgFbgemmLinearInt8Weight(ptr *Ctensor, input Ctensor, weight Ctensor, packed Ctensor, ...)
func AtgFbgemmLinearInt8WeightFp32Activation(ptr *Ctensor, input Ctensor, weight Ctensor, packed Ctensor, ...)
func AtgFbgemmPackGemmMatrixFp16(ptr *Ctensor, input Ctensor)
func AtgFbgemmPackQuantizedMatrix(ptr *Ctensor, input Ctensor)
func AtgFbgemmPackQuantizedMatrixKn(ptr *Ctensor, input Ctensor, k int64, n int64)
func AtgFeatureAlphaDropout(ptr *Ctensor, input Ctensor, p float64, train int32)
func AtgFeatureAlphaDropout_(ptr *Ctensor, self Ctensor, p float64, train int32)
func AtgFeatureDropout(ptr *Ctensor, input Ctensor, p float64, train int32)
func AtgFeatureDropout_(ptr *Ctensor, self Ctensor, p float64, train int32)
func AtgFftFft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)
func AtgFftFft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftFft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftFftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, ...)
func AtgFftFftfreq(ptr *Ctensor, n int64, d float64, optionsKind int32, optionsDevice int32)
func AtgFftFftfreqOut(ptr *Ctensor, out Ctensor, n int64, d float64)
func AtgFftFftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftFftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftFftshift(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)
func AtgFftHfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)
func AtgFftHfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftHfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftHfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, ...)
func AtgFftHfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftHfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftIfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)
func AtgFftIfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftIfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftIfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, ...)
func AtgFftIfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftIfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftIfftshift(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)
func AtgFftIhfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)
func AtgFftIhfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftIhfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftIhfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, ...)
func AtgFftIhfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftIhfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftIrfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)
func AtgFftIrfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftIrfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftIrfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, ...)
func AtgFftIrfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftIrfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftRfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)
func AtgFftRfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftRfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFftRfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, ...)
func AtgFftRfftfreq(ptr *Ctensor, n int64, d float64, optionsKind int32, optionsDevice int32)
func AtgFftRfftfreqOut(ptr *Ctensor, out Ctensor, n int64, d float64)
func AtgFftRfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, ...)
func AtgFftRfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, ...)
func AtgFill(ptr *Ctensor, self Ctensor, value Cscalar)
func AtgFillDiagonal_(ptr *Ctensor, self Ctensor, fillValue Cscalar, wrap int32)
func AtgFillScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, value Cscalar)
func AtgFillTensor(ptr *Ctensor, self Ctensor, value Ctensor)
func AtgFillTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, value Ctensor)
func AtgFillTensor_(ptr *Ctensor, self Ctensor, value Ctensor)
func AtgFill_(ptr *Ctensor, self Ctensor, value Cscalar)
func AtgFix(ptr *Ctensor, self Ctensor)
func AtgFixOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgFix_(ptr *Ctensor, self Ctensor)
func AtgFlatten(ptr *Ctensor, self Ctensor, startDim int64, endDim int64)
func AtgFlattenDenseTensors(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgFlip(ptr *Ctensor, self Ctensor, dimsData []int64, dimsLen int)
func AtgFlipOut(ptr *Ctensor, out Ctensor, self Ctensor, dimsData []int64, dimsLen int)
func AtgFliplr(ptr *Ctensor, self Ctensor)
func AtgFlipud(ptr *Ctensor, self Ctensor)
func AtgFloatPower(ptr *Ctensor, self Ctensor, exponent Ctensor)
func AtgFloatPowerScalar(ptr *Ctensor, selfScalar Cscalar, exponent Ctensor)
func AtgFloatPowerScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, exponent Ctensor)
func AtgFloatPowerTensorScalar(ptr *Ctensor, self Ctensor, exponent Cscalar)
func AtgFloatPowerTensorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, exponent Cscalar)
func AtgFloatPowerTensorTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, exponent Ctensor)
func AtgFloatPowerTensor_(ptr *Ctensor, self Ctensor, exponent Ctensor)
func AtgFloatPower_(ptr *Ctensor, self Ctensor, exponent Cscalar)
func AtgFloor(ptr *Ctensor, self Ctensor)
func AtgFloorDivide(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgFloorDivideOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgFloorDivideScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgFloorDivideScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgFloorDivide_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgFloorOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgFloor_(ptr *Ctensor, self Ctensor)
func AtgFmax(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgFmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgFmin(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgFminOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgFmod(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgFmodScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgFmodTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgFmodTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgFmodTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgFmod_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgFrac(ptr *Ctensor, self Ctensor)
func AtgFracOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgFrac_(ptr *Ctensor, self Ctensor)
func AtgFractionalMaxPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgFractionalMaxPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgFractionalMaxPool2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgFractionalMaxPool2dOutput(ptr *Ctensor, output Ctensor, indices Ctensor, self Ctensor, ...)
func AtgFractionalMaxPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgFractionalMaxPool3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgFractionalMaxPool3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgFractionalMaxPool3dOutput(ptr *Ctensor, output Ctensor, indices Ctensor, self Ctensor, ...)
func AtgFrexp(ptr *Ctensor, self Ctensor)
func AtgFrexpTensorOut(ptr *Ctensor, mantissa Ctensor, exponent Ctensor, self Ctensor)
func AtgFrobeniusNorm(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)
func AtgFrobeniusNormOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgFromFile(ptr *Ctensor, filename string, shared int32, sizeVal int64, sizeNull int, ...)
func AtgFromFileOut(ptr *Ctensor, out Ctensor, filename string, shared int32, sizeVal int64, ...)
func AtgFull(ptr *Ctensor, sizeData []int64, sizeLen int, fillValue Cscalar, ...)
func AtgFullLike(ptr *Ctensor, self Ctensor, fillValue Cscalar)
func AtgFullLikeOut(ptr *Ctensor, out Ctensor, self Ctensor, fillValue Cscalar)
func AtgFullOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, fillValue Cscalar)
func AtgFusedMovingAvgObsFakeQuant(ptr *Ctensor, self Ctensor, observerOn Ctensor, fakeQuantOn Ctensor, ...)
func AtgGather(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, sparseGrad int32)
func AtgGatherBackward(ptr *Ctensor, grad Ctensor, self Ctensor, dim int64, index Ctensor, ...)
func AtgGatherOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, ...)
func AtgGcd(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGcdOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgGcd_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGe(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgGeScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgGeTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGeTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgGeTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGe_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgGelu(ptr *Ctensor, self Ctensor, approximate string)
func AtgGeluBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, approximate string)
func AtgGeluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgGeluOut(ptr *Ctensor, out Ctensor, self Ctensor, approximate string)
func AtgGelu_(ptr *Ctensor, self Ctensor, approximate string)
func AtgGeometric(ptr *Ctensor, self Ctensor, p float64)
func AtgGeometricOut(ptr *Ctensor, out Ctensor, self Ctensor, p float64)
func AtgGeometric_(ptr *Ctensor, self Ctensor, p float64)
func AtgGeqrf(ptr *Ctensor, self Ctensor)
func AtgGeqrfA(ptr *Ctensor, a Ctensor, tau Ctensor, self Ctensor)
func AtgGer(ptr *Ctensor, self Ctensor, vec2 Ctensor)
func AtgGerOut(ptr *Ctensor, out Ctensor, self Ctensor, vec2 Ctensor)
func AtgGlu(ptr *Ctensor, self Ctensor, dim int64)
func AtgGluBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, dim int64)
func AtgGluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, dim int64)
func AtgGluBackwardJvp(ptr *Ctensor, gradX Ctensor, gradGlu Ctensor, x Ctensor, dgradGlu Ctensor, ...)
func AtgGluBackwardJvpOut(ptr *Ctensor, out Ctensor, gradX Ctensor, gradGlu Ctensor, x Ctensor, ...)
func AtgGluJvp(ptr *Ctensor, glu Ctensor, x Ctensor, dx Ctensor, dim int64)
func AtgGluJvpOut(ptr *Ctensor, out Ctensor, glu Ctensor, x Ctensor, dx Ctensor, dim int64)
func AtgGluOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)
func AtgGrad(ptr *Ctensor, self Ctensor)
func AtgGreater(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgGreaterEqual(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgGreaterEqualScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgGreaterEqualTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGreaterEqualTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgGreaterEqualTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGreaterEqual_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgGreaterScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgGreaterTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGreaterTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgGreaterTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGreater_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgGridSampler(ptr *Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, ...)
func AtgGridSampler2d(ptr *Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, ...)
func AtgGridSampler2dOut(ptr *Ctensor, out Ctensor, input Ctensor, grid Ctensor, ...)
func AtgGridSampler3d(ptr *Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, ...)
func AtgGridSampler3dOut(ptr *Ctensor, out Ctensor, input Ctensor, grid Ctensor, ...)
func AtgGroupNorm(ptr *Ctensor, input Ctensor, numGroups int64, weight Ctensor, bias Ctensor, ...)
func AtgGru(ptr *Ctensor, input Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, ...)
func AtgGruCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, ...)
func AtgGruData(ptr *Ctensor, data Ctensor, batchSizes Ctensor, hx Ctensor, ...)
func AtgGt(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgGtScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgGtTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGtTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgGtTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgGt_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgHammingWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)
func AtgHammingWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)
func AtgHammingWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, ...)
func AtgHammingWindowPeriodicAlpha(ptr *Ctensor, windowLength int64, periodic int32, alpha float64, ...)
func AtgHammingWindowPeriodicAlphaBeta(ptr *Ctensor, windowLength int64, periodic int32, alpha float64, beta float64, ...)
func AtgHammingWindowPeriodicAlphaBetaOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32, alpha float64, ...)
func AtgHammingWindowPeriodicAlphaOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32, alpha float64)
func AtgHammingWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)
func AtgHannWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)
func AtgHannWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)
func AtgHannWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, ...)
func AtgHannWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)
func AtgHardshrink(ptr *Ctensor, self Ctensor)
func AtgHardshrinkBackward(ptr *Ctensor, gradOut Ctensor, self Ctensor, lambd Cscalar)
func AtgHardshrinkBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOut Ctensor, self Ctensor, lambd Cscalar)
func AtgHardshrinkOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgHardsigmoid(ptr *Ctensor, self Ctensor)
func AtgHardsigmoidBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)
func AtgHardsigmoidBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor)
func AtgHardsigmoidOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgHardsigmoid_(ptr *Ctensor, self Ctensor)
func AtgHardswish(ptr *Ctensor, self Ctensor)
func AtgHardswishBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)
func AtgHardswishBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor)
func AtgHardswishOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgHardswish_(ptr *Ctensor, self Ctensor)
func AtgHardtanh(ptr *Ctensor, self Ctensor)
func AtgHardtanhBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, minVal Cscalar, maxVal Cscalar)
func AtgHardtanhBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgHardtanhOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgHardtanh_(ptr *Ctensor, self Ctensor)
func AtgHeaviside(ptr *Ctensor, self Ctensor, values Ctensor)
func AtgHeavisideOut(ptr *Ctensor, out Ctensor, self Ctensor, values Ctensor)
func AtgHeaviside_(ptr *Ctensor, self Ctensor, values Ctensor)
func AtgHingeEmbeddingLoss(ptr *Ctensor, self Ctensor, target Ctensor, margin float64, reduction int64)
func AtgHistc(ptr *Ctensor, self Ctensor, bins int64)
func AtgHistcOut(ptr *Ctensor, out Ctensor, self Ctensor, bins int64)
func AtgHspmm(ptr *Ctensor, mat1 Ctensor, mat2 Ctensor)
func AtgHspmmOut(ptr *Ctensor, out Ctensor, mat1 Ctensor, mat2 Ctensor)
func AtgHstack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgHstackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgHuberLoss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64, delta float64)
func AtgHuberLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, ...)
func AtgHuberLossBackwardOut(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgHuberLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64, ...)
func AtgHypot(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgHypotOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgHypot_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgI0(ptr *Ctensor, self Ctensor)
func AtgI0Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgI0_(ptr *Ctensor, self Ctensor)
func AtgIgamma(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgIgammaOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgIgamma_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgIgammac(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgIgammacOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgIgammac_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgIm2col(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgIm2colOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgImag(ptr *Ctensor, self Ctensor)
func AtgIndex(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int)
func AtgIndexAdd(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)
func AtgIndexAddOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, ...)
func AtgIndexAdd_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)
func AtgIndexCopy(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)
func AtgIndexCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, ...)
func AtgIndexCopy_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)
func AtgIndexFill(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)
func AtgIndexFillIntScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, ...)
func AtgIndexFillIntTensor(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Ctensor)
func AtgIndexFillIntTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, ...)
func AtgIndexFillIntTensor_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Ctensor)
func AtgIndexFill_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)
func AtgIndexPut(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, ...)
func AtgIndexPutOut(ptr *Ctensor, out Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, ...)
func AtgIndexPut_(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, ...)
func AtgIndexReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor, ...)
func AtgIndexReduceOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, ...)
func AtgIndexReduce_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor, ...)
func AtgIndexSelect(ptr *Ctensor, self Ctensor, dim int64, index Ctensor)
func AtgIndexSelectBackward(ptr *Ctensor, grad Ctensor, selfSizesData []int64, selfSizesLen int, dim int64, ...)
func AtgIndexSelectOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor)
func AtgIndexTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int)
func AtgIndices(ptr *Ctensor, self Ctensor)
func AtgIndicesCopy(ptr *Ctensor, self Ctensor)
func AtgIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgInfinitelyDifferentiableGeluBackward(ptr *Ctensor, grad Ctensor, self Ctensor)
func AtgInner(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgInnerOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgInstanceNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, ...)
func AtgIntRepr(ptr *Ctensor, self Ctensor)
func AtgIntReprOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgInverse(ptr *Ctensor, self Ctensor)
func AtgInverseOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgIsCoalesced(self Ctensor) bool
func AtgIsComplex(self Ctensor) bool
func AtgIsConj(self Ctensor) bool
func AtgIsDistributed(self Ctensor) bool
func AtgIsFloatingPoint(self Ctensor) bool
func AtgIsInference(self Ctensor) bool
func AtgIsLeaf(self Ctensor) bool
func AtgIsNeg(self Ctensor) bool
func AtgIsNonzero(self Ctensor) bool
func AtgIsPinned(self Ctensor, device int32) bool
func AtgIsSameSize(self Ctensor, other Ctensor) bool
func AtgIsSetTo(self Ctensor, tensor Ctensor) bool
func AtgIsSigned(self Ctensor) bool
func AtgIsVulkanAvailable() bool
func AtgIsclose(ptr *Ctensor, self Ctensor, other Ctensor, rtol float64, atol float64, ...)
func AtgIsfinite(ptr *Ctensor, self Ctensor)
func AtgIsin(ptr *Ctensor, elements Ctensor, testElements Ctensor, assumeUnique int32, ...)
func AtgIsinScalarTensor(ptr *Ctensor, element Cscalar, testElements Ctensor, assumeUnique int32, ...)
func AtgIsinScalarTensorOut(ptr *Ctensor, out Ctensor, element Cscalar, testElements Ctensor, ...)
func AtgIsinTensorScalar(ptr *Ctensor, elements Ctensor, testElement Cscalar, assumeUnique int32, ...)
func AtgIsinTensorScalarOut(ptr *Ctensor, out Ctensor, elements Ctensor, testElement Cscalar, ...)
func AtgIsinTensorTensorOut(ptr *Ctensor, out Ctensor, elements Ctensor, testElements Ctensor, ...)
func AtgIsinf(ptr *Ctensor, self Ctensor)
func AtgIsinfOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgIsnan(ptr *Ctensor, self Ctensor)
func AtgIsnanOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgIsneginf(ptr *Ctensor, self Ctensor)
func AtgIsneginfOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgIsposinf(ptr *Ctensor, self Ctensor)
func AtgIsposinfOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgIsreal(ptr *Ctensor, self Ctensor)
func AtgIstft(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, ...)
func AtgKaiserWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)
func AtgKaiserWindowBeta(ptr *Ctensor, windowLength int64, periodic int32, beta float64, ...)
func AtgKaiserWindowBetaOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32, beta float64)
func AtgKaiserWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)
func AtgKaiserWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, ...)
func AtgKaiserWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)
func AtgKlDiv(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64, logTarget int32)
func AtgKron(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgKronOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgKthvalue(ptr *Ctensor, self Ctensor, k int64, dim int64, keepdim int32)
func AtgKthvalueValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, k int64, ...)
func AtgL1Loss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64)
func AtgLayerNorm(ptr *Ctensor, input Ctensor, normalizedShapeData []int64, ...)
func AtgLcm(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLcmOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLcm_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLdexp(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLdexpOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLdexp_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLe(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgLeScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgLeTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLeTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLeTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLe_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgLeakyRelu(ptr *Ctensor, self Ctensor)
func AtgLeakyReluBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, negativeSlope Cscalar, ...)
func AtgLeakyReluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgLeakyReluOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLeakyRelu_(ptr *Ctensor, self Ctensor)
func AtgLerp(ptr *Ctensor, self Ctensor, end Ctensor, weight Cscalar)
func AtgLerpScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, end Ctensor, weight Cscalar)
func AtgLerpTensor(ptr *Ctensor, self Ctensor, end Ctensor, weight Ctensor)
func AtgLerpTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, end Ctensor, weight Ctensor)
func AtgLerpTensor_(ptr *Ctensor, self Ctensor, end Ctensor, weight Ctensor)
func AtgLerp_(ptr *Ctensor, self Ctensor, end Ctensor, weight Cscalar)
func AtgLess(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgLessEqual(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgLessEqualScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgLessEqualTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLessEqualTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLessEqualTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLessEqual_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgLessScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgLessTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLessTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLessTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLess_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgLgamma(ptr *Ctensor, self Ctensor)
func AtgLgammaOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLgamma_(ptr *Ctensor, self Ctensor)
func AtgLift(ptr *Ctensor, self Ctensor)
func AtgLiftFresh(ptr *Ctensor, self Ctensor)
func AtgLiftFreshCopy(ptr *Ctensor, self Ctensor)
func AtgLiftFreshCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLiftOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLinalgCholesky(ptr *Ctensor, self Ctensor, upper int32)
func AtgLinalgCholeskyEx(ptr *Ctensor, self Ctensor, upper int32, checkErrors int32)
func AtgLinalgCholeskyExL(ptr *Ctensor, l Ctensor, info Ctensor, self Ctensor, upper int32, ...)
func AtgLinalgCholeskyOut(ptr *Ctensor, out Ctensor, self Ctensor, upper int32)
func AtgLinalgCond(ptr *Ctensor, self Ctensor, p Cscalar)
func AtgLinalgCondOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar)
func AtgLinalgCondPStr(ptr *Ctensor, self Ctensor, p string)
func AtgLinalgCondPStrOut(ptr *Ctensor, out Ctensor, self Ctensor, p string)
func AtgLinalgCross(ptr *Ctensor, self Ctensor, other Ctensor, dim int64)
func AtgLinalgCrossOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, dim int64)
func AtgLinalgDet(ptr *Ctensor, a Ctensor)
func AtgLinalgDetOut(ptr *Ctensor, out Ctensor, a Ctensor)
func AtgLinalgDiagonal(ptr *Ctensor, a Ctensor, offset int64, dim1 int64, dim2 int64)
func AtgLinalgEig(ptr *Ctensor, self Ctensor)
func AtgLinalgEigOut(ptr *Ctensor, eigenvalues Ctensor, eigenvectors Ctensor, self Ctensor)
func AtgLinalgEigh(ptr *Ctensor, self Ctensor, uPLO string)
func AtgLinalgEighEigvals(ptr *Ctensor, eigvals Ctensor, eigvecs Ctensor, self Ctensor, uPLO string)
func AtgLinalgEigvals(ptr *Ctensor, self Ctensor)
func AtgLinalgEigvalsOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLinalgEigvalsh(ptr *Ctensor, self Ctensor, uPLO string)
func AtgLinalgEigvalshOut(ptr *Ctensor, out Ctensor, self Ctensor, uPLO string)
func AtgLinalgHouseholderProduct(ptr *Ctensor, input Ctensor, tau Ctensor)
func AtgLinalgHouseholderProductOut(ptr *Ctensor, out Ctensor, input Ctensor, tau Ctensor)
func AtgLinalgInv(ptr *Ctensor, a Ctensor)
func AtgLinalgInvEx(ptr *Ctensor, a Ctensor, checkErrors int32)
func AtgLinalgInvExInverse(ptr *Ctensor, inverse Ctensor, info Ctensor, a Ctensor, checkErrors int32)
func AtgLinalgInvOut(ptr *Ctensor, out Ctensor, a Ctensor)
func AtgLinalgLdlFactor(ptr *Ctensor, self Ctensor, hermitian int32)
func AtgLinalgLdlFactorEx(ptr *Ctensor, self Ctensor, hermitian int32, checkErrors int32)
func AtgLinalgLdlFactorExOut(ptr *Ctensor, lD Ctensor, pivots Ctensor, info Ctensor, self Ctensor, ...)
func AtgLinalgLdlFactorOut(ptr *Ctensor, lD Ctensor, pivots Ctensor, self Ctensor, hermitian int32)
func AtgLinalgLdlSolve(ptr *Ctensor, lD Ctensor, pivots Ctensor, b Ctensor, hermitian int32)
func AtgLinalgLdlSolveOut(ptr *Ctensor, out Ctensor, lD Ctensor, pivots Ctensor, b Ctensor, ...)
func AtgLinalgLstsq(ptr *Ctensor, self Ctensor, b Ctensor, rcondVal float64, rcondNull int, ...)
func AtgLinalgLstsqOut(ptr *Ctensor, solution Ctensor, residuals Ctensor, rank Ctensor, ...)
func AtgLinalgLu(ptr *Ctensor, a Ctensor, pivot int32)
func AtgLinalgLuFactor(ptr *Ctensor, a Ctensor, pivot int32)
func AtgLinalgLuFactorEx(ptr *Ctensor, a Ctensor, pivot int32, checkErrors int32)
func AtgLinalgLuFactorExOut(ptr *Ctensor, lU Ctensor, pivots Ctensor, info Ctensor, a Ctensor, pivot int32, ...)
func AtgLinalgLuFactorOut(ptr *Ctensor, lU Ctensor, pivots Ctensor, a Ctensor, pivot int32)
func AtgLinalgLuOut(ptr *Ctensor, p Ctensor, l Ctensor, u Ctensor, a Ctensor, pivot int32)
func AtgLinalgLuSolve(ptr *Ctensor, lU Ctensor, pivots Ctensor, b Ctensor, left int32, adjoint int32)
func AtgLinalgLuSolveOut(ptr *Ctensor, out Ctensor, lU Ctensor, pivots Ctensor, b Ctensor, left int32, ...)
func AtgLinalgMatmul(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLinalgMatmulOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLinalgMatrixExp(ptr *Ctensor, self Ctensor)
func AtgLinalgMatrixExpOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLinalgMatrixPower(ptr *Ctensor, self Ctensor, n int64)
func AtgLinalgMatrixPowerOut(ptr *Ctensor, out Ctensor, self Ctensor, n int64)
func AtgLinalgMatrixRank(ptr *Ctensor, self Ctensor, tol float64, hermitian int32)
func AtgLinalgMatrixRankAtolRtolFloat(ptr *Ctensor, self Ctensor, atolVal float64, atolNull int, rtolVal float64, ...)
func AtgLinalgMatrixRankAtolRtolFloatOut(ptr *Ctensor, out Ctensor, self Ctensor, atolVal float64, atolNull int, ...)
func AtgLinalgMatrixRankAtolRtolTensor(ptr *Ctensor, input Ctensor, atol Ctensor, rtol Ctensor, hermitian int32)
func AtgLinalgMatrixRankAtolRtolTensorOut(ptr *Ctensor, out Ctensor, input Ctensor, atol Ctensor, rtol Ctensor, ...)
func AtgLinalgMatrixRankOut(ptr *Ctensor, out Ctensor, self Ctensor, tol float64, hermitian int32)
func AtgLinalgMatrixRankOutTolTensor(ptr *Ctensor, out Ctensor, input Ctensor, tol Ctensor, hermitian int32)
func AtgLinalgMatrixRankTolTensor(ptr *Ctensor, input Ctensor, tol Ctensor, hermitian int32)
func AtgLinalgMultiDot(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgLinalgMultiDotOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgLinalgNorm(ptr *Ctensor, self Ctensor, ord Cscalar, dimData []int64, dimLen int, ...)
func AtgLinalgNormOrdStr(ptr *Ctensor, self Ctensor, ord string, dimData []int64, dimLen int, ...)
func AtgLinalgNormOrdStrOut(ptr *Ctensor, out Ctensor, self Ctensor, ord string, dimData []int64, ...)
func AtgLinalgNormOut(ptr *Ctensor, out Ctensor, self Ctensor, ord Cscalar, dimData []int64, ...)
func AtgLinalgPinv(ptr *Ctensor, self Ctensor, rcond float64, hermitian int32)
func AtgLinalgPinvAtolRtolFloat(ptr *Ctensor, self Ctensor, atolVal float64, atolNull int, rtolVal float64, ...)
func AtgLinalgPinvAtolRtolFloatOut(ptr *Ctensor, out Ctensor, self Ctensor, atolVal float64, atolNull int, ...)
func AtgLinalgPinvAtolRtolTensor(ptr *Ctensor, self Ctensor, atol Ctensor, rtol Ctensor, hermitian int32)
func AtgLinalgPinvAtolRtolTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, atol Ctensor, rtol Ctensor, ...)
func AtgLinalgPinvOut(ptr *Ctensor, out Ctensor, self Ctensor, rcond float64, hermitian int32)
func AtgLinalgPinvOutRcondTensor(ptr *Ctensor, out Ctensor, self Ctensor, rcond Ctensor, hermitian int32)
func AtgLinalgPinvRcondTensor(ptr *Ctensor, self Ctensor, rcond Ctensor, hermitian int32)
func AtgLinalgQr(ptr *Ctensor, a Ctensor, mode string)
func AtgLinalgQrOut(ptr *Ctensor, q Ctensor, r Ctensor, a Ctensor, mode string)
func AtgLinalgSlogdet(ptr *Ctensor, a Ctensor)
func AtgLinalgSlogdetOut(ptr *Ctensor, sign Ctensor, logabsdet Ctensor, a Ctensor)
func AtgLinalgSolve(ptr *Ctensor, a Ctensor, b Ctensor, left int32)
func AtgLinalgSolveEx(ptr *Ctensor, a Ctensor, b Ctensor, left int32, checkErrors int32)
func AtgLinalgSolveExOut(ptr *Ctensor, result Ctensor, info Ctensor, a Ctensor, b Ctensor, left int32, ...)
func AtgLinalgSolveOut(ptr *Ctensor, out Ctensor, a Ctensor, b Ctensor, left int32)
func AtgLinalgSolveTriangular(ptr *Ctensor, self Ctensor, b Ctensor, upper int32, left int32, ...)
func AtgLinalgSolveTriangularOut(ptr *Ctensor, out Ctensor, self Ctensor, b Ctensor, upper int32, left int32, ...)
func AtgLinalgSvd(ptr *Ctensor, a Ctensor, fullMatrices int32, driver string)
func AtgLinalgSvdU(ptr *Ctensor, u Ctensor, s Ctensor, vh Ctensor, a Ctensor, fullMatrices int32, ...)
func AtgLinalgSvdvals(ptr *Ctensor, a Ctensor, driver string)
func AtgLinalgSvdvalsOut(ptr *Ctensor, out Ctensor, a Ctensor, driver string)
func AtgLinalgTensorinv(ptr *Ctensor, self Ctensor, ind int64)
func AtgLinalgTensorinvOut(ptr *Ctensor, out Ctensor, self Ctensor, ind int64)
func AtgLinalgTensorsolve(ptr *Ctensor, self Ctensor, other Ctensor, dimsData []int64, dimsLen int)
func AtgLinalgTensorsolveOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, dimsData []int64, ...)
func AtgLinalgVander(ptr *Ctensor, x Ctensor, nVal int64, nNull int)
func AtgLinalgVecdot(ptr *Ctensor, x Ctensor, y Ctensor, dim int64)
func AtgLinalgVecdotOut(ptr *Ctensor, out Ctensor, x Ctensor, y Ctensor, dim int64)
func AtgLinear(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor)
func AtgLinearOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor)
func AtgLinspace(ptr *Ctensor, start Cscalar, end Cscalar, steps int64, optionsKind int32, ...)
func AtgLinspaceOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar, steps int64)
func AtgLog(ptr *Ctensor, self Ctensor)
func AtgLog10(ptr *Ctensor, self Ctensor)
func AtgLog10Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLog10_(ptr *Ctensor, self Ctensor)
func AtgLog1p(ptr *Ctensor, self Ctensor)
func AtgLog1pOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLog1p_(ptr *Ctensor, self Ctensor)
func AtgLog2(ptr *Ctensor, self Ctensor)
func AtgLog2Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLog2_(ptr *Ctensor, self Ctensor)
func AtgLogNormal(ptr *Ctensor, self Ctensor, mean float64, std float64)
func AtgLogNormalOut(ptr *Ctensor, out Ctensor, self Ctensor, mean float64, std float64)
func AtgLogNormal_(ptr *Ctensor, self Ctensor, mean float64, std float64)
func AtgLogOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLogSigmoid(ptr *Ctensor, self Ctensor)
func AtgLogSigmoidBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, buffer Ctensor)
func AtgLogSigmoidBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgLogSigmoidOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLogSoftmax(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func AtgLogSoftmaxIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, dtype int32)
func AtgLog_(ptr *Ctensor, self Ctensor)
func AtgLogaddexp(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLogaddexp2(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLogaddexp2Out(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLogaddexpOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLogcumsumexp(ptr *Ctensor, self Ctensor, dim int64)
func AtgLogcumsumexpOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)
func AtgLogdet(ptr *Ctensor, self Ctensor)
func AtgLogicalAnd(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLogicalAndOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLogicalAnd_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLogicalNot(ptr *Ctensor, self Ctensor)
func AtgLogicalNotOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgLogicalNot_(ptr *Ctensor, self Ctensor)
func AtgLogicalOr(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLogicalOrOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLogicalOr_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLogicalXor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLogicalXorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLogicalXor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLogit(ptr *Ctensor, self Ctensor, epsVal float64, epsNull int)
func AtgLogitBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, epsVal float64, epsNull int)
func AtgLogitBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgLogitOut(ptr *Ctensor, out Ctensor, self Ctensor, epsVal float64, epsNull int)
func AtgLogit_(ptr *Ctensor, self Ctensor, epsVal float64, epsNull int)
func AtgLogspace(ptr *Ctensor, start Cscalar, end Cscalar, steps int64, base float64, ...)
func AtgLogspaceOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar, steps int64, ...)
func AtgLogsumexp(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)
func AtgLogsumexpOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgLstm(ptr *Ctensor, input Ctensor, hxData []Ctensor, hxLen int, paramsData []Ctensor, ...)
func AtgLstmCell(ptr *Ctensor, input Ctensor, hxData []Ctensor, hxLen int, wIh Ctensor, ...)
func AtgLstmData(ptr *Ctensor, data Ctensor, batchSizes Ctensor, hxData []Ctensor, hxLen int, ...)
func AtgLt(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgLtScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgLtTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLtTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgLtTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgLt_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgLuSolve(ptr *Ctensor, self Ctensor, lUData Ctensor, lUPivots Ctensor)
func AtgLuSolveOut(ptr *Ctensor, out Ctensor, self Ctensor, lUData Ctensor, lUPivots Ctensor)
func AtgLuUnpack(ptr *Ctensor, lUData Ctensor, lUPivots Ctensor, unpackData int32, ...)
func AtgLuUnpackOut(ptr *Ctensor, p Ctensor, l Ctensor, u Ctensor, lUData Ctensor, ...)
func AtgMarginRankingLoss(ptr *Ctensor, input1 Ctensor, input2 Ctensor, target Ctensor, margin float64, ...)
func AtgMaskedFill(ptr *Ctensor, self Ctensor, mask Ctensor, value Cscalar)
func AtgMaskedFillScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, value Cscalar)
func AtgMaskedFillTensor(ptr *Ctensor, self Ctensor, mask Ctensor, value Ctensor)
func AtgMaskedFillTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, value Ctensor)
func AtgMaskedFillTensor_(ptr *Ctensor, self Ctensor, mask Ctensor, value Ctensor)
func AtgMaskedFill_(ptr *Ctensor, self Ctensor, mask Ctensor, value Cscalar)
func AtgMaskedScatter(ptr *Ctensor, self Ctensor, mask Ctensor, source Ctensor)
func AtgMaskedScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, source Ctensor)
func AtgMaskedScatter_(ptr *Ctensor, self Ctensor, mask Ctensor, source Ctensor)
func AtgMaskedSelect(ptr *Ctensor, self Ctensor, mask Ctensor)
func AtgMaskedSelectBackward(ptr *Ctensor, grad Ctensor, input Ctensor, mask Ctensor)
func AtgMaskedSelectOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor)
func AtgMatmul(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgMatmulOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgMatrixExp(ptr *Ctensor, self Ctensor)
func AtgMatrixExpBackward(ptr *Ctensor, self Ctensor, grad Ctensor)
func AtgMatrixH(ptr *Ctensor, self Ctensor)
func AtgMatrixPower(ptr *Ctensor, self Ctensor, n int64)
func AtgMatrixPowerOut(ptr *Ctensor, out Ctensor, self Ctensor, n int64)
func AtgMax(ptr *Ctensor, self Ctensor)
func AtgMaxDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)
func AtgMaxDimMax(ptr *Ctensor, max Ctensor, maxValues Ctensor, self Ctensor, dim int64, ...)
func AtgMaxOther(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgMaxOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgMaxPool1d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgMaxPool1dWithIndices(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgMaxPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgMaxPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgMaxPool2dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgMaxPool2dWithIndices(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgMaxPool2dWithIndicesBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgMaxPool2dWithIndicesBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgMaxPool2dWithIndicesOut(ptr *Ctensor, out Ctensor, indices Ctensor, self Ctensor, ...)
func AtgMaxPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgMaxPool3dWithIndices(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgMaxPool3dWithIndicesBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgMaxPool3dWithIndicesBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgMaxPool3dWithIndicesOut(ptr *Ctensor, out Ctensor, indices Ctensor, self Ctensor, ...)
func AtgMaxUnaryOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgMaxUnpool2d(ptr *Ctensor, self Ctensor, indices Ctensor, outputSizeData []int64, ...)
func AtgMaxUnpool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, indices Ctensor, ...)
func AtgMaxUnpool3d(ptr *Ctensor, self Ctensor, indices Ctensor, outputSizeData []int64, ...)
func AtgMaxUnpool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, indices Ctensor, ...)
func AtgMaximum(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgMaximumOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgMean(ptr *Ctensor, self Ctensor, dtype int32)
func AtgMeanDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, ...)
func AtgMeanOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgMedian(ptr *Ctensor, self Ctensor)
func AtgMedianDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)
func AtgMedianDimValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64, ...)
func AtgMedianOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgMh(ptr *Ctensor, self Ctensor)
func AtgMin(ptr *Ctensor, self Ctensor)
func AtgMinDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)
func AtgMinDimMin(ptr *Ctensor, min Ctensor, minIndices Ctensor, self Ctensor, dim int64, ...)
func AtgMinOther(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgMinOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgMinUnaryOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgMinimum(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgMinimumOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgMiopenBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, ...)
func AtgMiopenBatchNormBackward(ptr *Ctensor, input Ctensor, gradOutput Ctensor, weight Ctensor, ...)
func AtgMiopenBatchNormBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, ...)
func AtgMiopenBatchNormOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, ...)
func AtgMiopenConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, ...)
func AtgMiopenConvolutionAddRelu(ptr *Ctensor, self Ctensor, weight Ctensor, z Ctensor, alpha Cscalar, ...)
func AtgMiopenConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgMiopenConvolutionRelu(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func AtgMiopenConvolutionTranspose(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, ...)
func AtgMiopenConvolutionTransposeOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgMiopenDepthwiseConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, ...)
func AtgMiopenDepthwiseConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgMiopenRnn(ptr *Ctensor, input Ctensor, weightData []Ctensor, weightLen int, ...)
func AtgMiopenRnnOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, ...)
func AtgMish(ptr *Ctensor, self Ctensor)
func AtgMishBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)
func AtgMishOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgMish_(ptr *Ctensor, self Ctensor)
func AtgMkldnnAdaptiveAvgPool2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)
func AtgMkldnnAdaptiveAvgPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)
func AtgMkldnnAdaptiveAvgPool2dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor)
func AtgMkldnnAdaptiveAvgPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgMkldnnConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, ...)
func AtgMkldnnConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgMkldnnLinear(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor)
func AtgMkldnnLinearBackwardInput(ptr *Ctensor, inputSizeData []int64, inputSizeLen int, gradOutput Ctensor, ...)
func AtgMkldnnLinearBackwardInputOut(ptr *Ctensor, out Ctensor, inputSizeData []int64, inputSizeLen int, ...)
func AtgMkldnnLinearBackwardWeights(ptr *Ctensor, gradOutput Ctensor, input Ctensor, weight Ctensor, ...)
func AtgMkldnnLinearBackwardWeightsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, gradOutput Ctensor, input Ctensor, ...)
func AtgMkldnnLinearOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor)
func AtgMkldnnMaxPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgMkldnnMaxPool2dBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor, input Ctensor, ...)
func AtgMkldnnMaxPool2dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, input Ctensor, ...)
func AtgMkldnnMaxPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgMkldnnMaxPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgMkldnnMaxPool3dBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor, input Ctensor, ...)
func AtgMkldnnMaxPool3dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, input Ctensor, ...)
func AtgMkldnnMaxPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgMkldnnReorderConv2dWeight(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int, ...)
func AtgMkldnnReorderConv2dWeightOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int, ...)
func AtgMkldnnReorderConv3dWeight(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int, ...)
func AtgMkldnnReorderConv3dWeightOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int, ...)
func AtgMkldnnRnnLayer(ptr *Ctensor, input Ctensor, weight0 Ctensor, weight1 Ctensor, weight2 Ctensor, ...)
func AtgMkldnnRnnLayerBackward(ptr *Ctensor, input Ctensor, weight1 Ctensor, weight2 Ctensor, weight3 Ctensor, ...)
func AtgMkldnnRnnLayerBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, ...)
func AtgMkldnnRnnLayerOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, ...)
func AtgMm(ptr *Ctensor, self Ctensor, mat2 Ctensor)
func AtgMmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor)
func AtgMode(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)
func AtgModeValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64, ...)
func AtgMoveaxis(ptr *Ctensor, self Ctensor, sourceData []int64, sourceLen int, ...)
func AtgMoveaxisInt(ptr *Ctensor, self Ctensor, source int64, destination int64)
func AtgMovedim(ptr *Ctensor, self Ctensor, sourceData []int64, sourceLen int, ...)
func AtgMovedimInt(ptr *Ctensor, self Ctensor, source int64, destination int64)
func AtgMseLoss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64)
func AtgMseLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, ...)
func AtgMseLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgMseLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64)
func AtgMsort(ptr *Ctensor, self Ctensor)
func AtgMsortOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgMt(ptr *Ctensor, self Ctensor)
func AtgMul(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgMulOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgMulScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgMulScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgMulScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgMul_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgMultiMarginLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, p Cscalar, ...)
func AtgMultiMarginLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgMultilabelMarginLoss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64)
func AtgMultilabelMarginLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, ...)
func AtgMultilabelMarginLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgMultilabelMarginLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64)
func AtgMultinomial(ptr *Ctensor, self Ctensor, numSamples int64, replacement int32)
func AtgMultinomialOut(ptr *Ctensor, out Ctensor, self Ctensor, numSamples int64, replacement int32)
func AtgMultiply(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgMultiplyOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgMultiplyScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgMultiplyScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgMultiply_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgMv(ptr *Ctensor, self Ctensor, vec Ctensor)
func AtgMvOut(ptr *Ctensor, out Ctensor, self Ctensor, vec Ctensor)
func AtgMvlgamma(ptr *Ctensor, self Ctensor, p int64)
func AtgMvlgammaOut(ptr *Ctensor, out Ctensor, self Ctensor, p int64)
func AtgMvlgamma_(ptr *Ctensor, self Ctensor, p int64)
func AtgNanToNum(ptr *Ctensor, self Ctensor, nanVal float64, nanNull int, posinfVal float64, ...)
func AtgNanToNumOut(ptr *Ctensor, out Ctensor, self Ctensor, nanVal float64, nanNull int, ...)
func AtgNanToNum_(ptr *Ctensor, self Ctensor, nanVal float64, nanNull int, posinfVal float64, ...)
func AtgNanmean(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, ...)
func AtgNanmeanOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgNanmedian(ptr *Ctensor, self Ctensor)
func AtgNanmedianDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)
func AtgNanmedianDimValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64, ...)
func AtgNanmedianOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgNanquantile(ptr *Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, ...)
func AtgNanquantileOut(ptr *Ctensor, out Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, ...)
func AtgNanquantileScalar(ptr *Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, ...)
func AtgNanquantileScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, ...)
func AtgNansum(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, ...)
func AtgNansumOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgNarrow(ptr *Ctensor, self Ctensor, dim int64, start int64, length int64)
func AtgNarrowCopy(ptr *Ctensor, self Ctensor, dim int64, start int64, length int64)
func AtgNarrowCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, start int64, length int64)
func AtgNarrowTensor(ptr *Ctensor, self Ctensor, dim int64, start Ctensor, length int64)
func AtgNativeBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, ...)
func AtgNativeBatchNormOut(ptr *Ctensor, out Ctensor, saveMean Ctensor, saveInvstd Ctensor, input Ctensor, ...)
func AtgNativeChannelShuffle(ptr *Ctensor, self Ctensor, groups int64)
func AtgNativeDropout(ptr *Ctensor, input Ctensor, p float64, train int32)
func AtgNativeDropoutBackward(ptr *Ctensor, gradOutput Ctensor, mask Ctensor, scale float64)
func AtgNativeDropoutBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, mask Ctensor, scale float64)
func AtgNativeDropoutOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, p float64, ...)
func AtgNativeGroupNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, n int64, c int64, ...)
func AtgNativeGroupNormOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, ...)
func AtgNativeLayerNorm(ptr *Ctensor, input Ctensor, normalizedShapeData []int64, ...)
func AtgNativeLayerNormOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, ...)
func AtgNativeNorm(ptr *Ctensor, self Ctensor)
func AtgNativeNormOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgNativeNormScalaroptDimDtype(ptr *Ctensor, self Ctensor, p Cscalar, dimData []int64, dimLen int, ...)
func AtgNativeNormScalaroptDimDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dimData []int64, ...)
func AtgNe(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgNeScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgNeTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgNeTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgNeTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgNe_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgNeg(ptr *Ctensor, self Ctensor)
func AtgNegOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgNeg_(ptr *Ctensor, self Ctensor)
func AtgNegative(ptr *Ctensor, self Ctensor)
func AtgNegativeOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgNegative_(ptr *Ctensor, self Ctensor)
func AtgNestedToPaddedTensor(ptr *Ctensor, self Ctensor, padding float64, outputSizeData []int64, ...)
func AtgNewEmpty(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgNewEmptyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgNewEmptyStrided(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, ...)
func AtgNewEmptyStridedOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, ...)
func AtgNewFull(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, fillValue Cscalar, ...)
func AtgNewFullOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, ...)
func AtgNewOnes(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgNewOnesOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgNewZeros(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgNewZerosOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgNextafter(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgNextafterOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgNextafter_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgNllLoss(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ...)
func AtgNllLoss2d(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ...)
func AtgNllLoss2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, weight Ctensor, ...)
func AtgNllLoss2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgNllLoss2dOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, weight Ctensor, ...)
func AtgNllLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, weight Ctensor, ...)
func AtgNllLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgNllLossNd(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ...)
func AtgNllLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, weight Ctensor, ...)
func AtgNonzero(ptr *Ctensor, self Ctensor)
func AtgNonzeroOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgNonzeroStatic(ptr *Ctensor, self Ctensor, size int64, fillValue int64)
func AtgNonzeroStaticOut(ptr *Ctensor, out Ctensor, self Ctensor, size int64, fillValue int64)
func AtgNorm(ptr *Ctensor, self Ctensor)
func AtgNormDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dimData []int64, ...)
func AtgNormExceptDim(ptr *Ctensor, v Ctensor, pow int64, dim int64)
func AtgNormOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dimData []int64, ...)
func AtgNormScalarOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgNormScalaroptDim(ptr *Ctensor, self Ctensor, p Cscalar, dimData []int64, dimLen int, ...)
func AtgNormScalaroptDimDtype(ptr *Ctensor, self Ctensor, p Cscalar, dimData []int64, dimLen int, ...)
func AtgNormScalaroptDtype(ptr *Ctensor, self Ctensor, p Cscalar, dtype int32)
func AtgNormScalaroptDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dtype int32)
func AtgNormalFunctional(ptr *Ctensor, self Ctensor, mean float64, std float64)
func AtgNormal_(ptr *Ctensor, self Ctensor, mean float64, std float64)
func AtgNotEqual(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgNotEqualScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgNotEqualTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgNotEqualTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgNotEqualTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgNotEqual_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgNuclearNorm(ptr *Ctensor, self Ctensor, keepdim int32)
func AtgNuclearNormDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)
func AtgNuclearNormDimOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgNuclearNormOut(ptr *Ctensor, out Ctensor, self Ctensor, keepdim int32)
func AtgNumpyT(ptr *Ctensor, self Ctensor)
func AtgOneHot(ptr *Ctensor, self Ctensor, numClasses int64)
func AtgOnes(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgOnesLike(ptr *Ctensor, self Ctensor)
func AtgOnesLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgOnesOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)
func AtgOrgqr(ptr *Ctensor, self Ctensor, input2 Ctensor)
func AtgOrgqrOut(ptr *Ctensor, out Ctensor, self Ctensor, input2 Ctensor)
func AtgOrmqr(ptr *Ctensor, self Ctensor, input2 Ctensor, input3 Ctensor, left int32, ...)
func AtgOrmqrOut(ptr *Ctensor, out Ctensor, self Ctensor, input2 Ctensor, input3 Ctensor, ...)
func AtgOuter(ptr *Ctensor, self Ctensor, vec2 Ctensor)
func AtgOuterOut(ptr *Ctensor, out Ctensor, self Ctensor, vec2 Ctensor)
func AtgOutputNr(self Ctensor) int64
func AtgPad(ptr *Ctensor, self Ctensor, padData []int64, padLen int, mode string, ...)
func AtgPadSequence(ptr *Ctensor, sequencesData []Ctensor, sequencesLen int, batchFirst int32, ...)
func AtgPairwiseDistance(ptr *Ctensor, x1 Ctensor, x2 Ctensor, p float64, eps float64, keepdim int32)
func AtgPdist(ptr *Ctensor, self Ctensor, p float64)
func AtgPermute(ptr *Ctensor, self Ctensor, dimsData []int64, dimsLen int)
func AtgPermuteCopy(ptr *Ctensor, self Ctensor, dimsData []int64, dimsLen int)
func AtgPermuteCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dimsData []int64, dimsLen int)
func AtgPinMemory(ptr *Ctensor, self Ctensor, device int32)
func AtgPinverse(ptr *Ctensor, self Ctensor, rcond float64)
func AtgPixelShuffle(ptr *Ctensor, self Ctensor, upscaleFactor int64)
func AtgPixelShuffleOut(ptr *Ctensor, out Ctensor, self Ctensor, upscaleFactor int64)
func AtgPixelUnshuffle(ptr *Ctensor, self Ctensor, downscaleFactor int64)
func AtgPixelUnshuffleOut(ptr *Ctensor, out Ctensor, self Ctensor, downscaleFactor int64)
func AtgPoisson(ptr *Ctensor, self Ctensor)
func AtgPoissonNllLoss(ptr *Ctensor, input Ctensor, target Ctensor, logInput int32, full int32, ...)
func AtgPoissonOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgPolar(ptr *Ctensor, abs Ctensor, angle Ctensor)
func AtgPolarOut(ptr *Ctensor, out Ctensor, abs Ctensor, angle Ctensor)
func AtgPolygamma(ptr *Ctensor, n int64, self Ctensor)
func AtgPolygammaOut(ptr *Ctensor, out Ctensor, n int64, self Ctensor)
func AtgPolygamma_(ptr *Ctensor, self Ctensor, n int64)
func AtgPositive(ptr *Ctensor, self Ctensor)
func AtgPow(ptr *Ctensor, self Ctensor, exponent Ctensor)
func AtgPowScalar(ptr *Ctensor, selfScalar Cscalar, exponent Ctensor)
func AtgPowScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, exponent Ctensor)
func AtgPowTensorScalar(ptr *Ctensor, self Ctensor, exponent Cscalar)
func AtgPowTensorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, exponent Cscalar)
func AtgPowTensorTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, exponent Ctensor)
func AtgPowTensor_(ptr *Ctensor, self Ctensor, exponent Ctensor)
func AtgPow_(ptr *Ctensor, self Ctensor, exponent Cscalar)
func AtgPrelu(ptr *Ctensor, self Ctensor, weight Ctensor)
func AtgProd(ptr *Ctensor, self Ctensor, dtype int32)
func AtgProdDimInt(ptr *Ctensor, self Ctensor, dim int64, keepdim int32, dtype int32)
func AtgProdIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, keepdim int32, dtype int32)
func AtgProdOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32)
func AtgPut(ptr *Ctensor, self Ctensor, index Ctensor, source Ctensor, accumulate int32)
func AtgPutOut(ptr *Ctensor, out Ctensor, self Ctensor, index Ctensor, source Ctensor, ...)
func AtgPut_(ptr *Ctensor, self Ctensor, index Ctensor, source Ctensor, accumulate int32)
func AtgQPerChannelAxis(self Ctensor) int64
func AtgQPerChannelScales(ptr *Ctensor, self Ctensor)
func AtgQPerChannelScalesOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgQPerChannelZeroPoints(ptr *Ctensor, self Ctensor)
func AtgQPerChannelZeroPointsOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgQScale(self Ctensor) float64
func AtgQZeroPoint(self Ctensor) int64
func AtgQr(ptr *Ctensor, self Ctensor, some int32)
func AtgQrQ(ptr *Ctensor, q Ctensor, r Ctensor, self Ctensor, some int32)
func AtgQuantile(ptr *Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, ...)
func AtgQuantileOut(ptr *Ctensor, out Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, ...)
func AtgQuantileScalar(ptr *Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, ...)
func AtgQuantileScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, ...)
func AtgQuantizePerChannel(ptr *Ctensor, self Ctensor, scales Ctensor, zeroPoints Ctensor, axis int64, ...)
func AtgQuantizePerChannelOut(ptr *Ctensor, out Ctensor, self Ctensor, scales Ctensor, zeroPoints Ctensor, ...)
func AtgQuantizePerTensor(ptr *Ctensor, self Ctensor, scale float64, zeroPoint int64, dtype int32)
func AtgQuantizePerTensorDynamic(ptr *Ctensor, self Ctensor, dtype int32, reduceRange int32)
func AtgQuantizePerTensorDynamicOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32, reduceRange int32)
func AtgQuantizePerTensorTensorQparams(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, dtype int32)
func AtgQuantizedBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, mean Ctensor, ...)
func AtgQuantizedBatchNormOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, ...)
func AtgQuantizedGruCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, ...)
func AtgQuantizedLstmCell(ptr *Ctensor, input Ctensor, hxData []Ctensor, hxLen int, wIh Ctensor, ...)
func AtgQuantizedMaxPool1d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgQuantizedMaxPool1dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgQuantizedMaxPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgQuantizedMaxPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgQuantizedMaxPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, ...)
func AtgQuantizedMaxPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, ...)
func AtgQuantizedRnnReluCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, ...)
func AtgQuantizedRnnTanhCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, ...)
func AtgRad2deg(ptr *Ctensor, self Ctensor)
func AtgRad2degOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgRad2deg_(ptr *Ctensor, self Ctensor)
func AtgRand(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgRandLike(ptr *Ctensor, self Ctensor)
func AtgRandLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgRandOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)
func AtgRandint(ptr *Ctensor, high int64, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgRandintLike(ptr *Ctensor, self Ctensor, high int64)
func AtgRandintLikeLowDtype(ptr *Ctensor, self Ctensor, low int64, high int64)
func AtgRandintLikeLowDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, low int64, high int64)
func AtgRandintLikeOut(ptr *Ctensor, out Ctensor, self Ctensor, high int64)
func AtgRandintLow(ptr *Ctensor, low int64, high int64, sizeData []int64, sizeLen int, ...)
func AtgRandintLowOut(ptr *Ctensor, out Ctensor, low int64, high int64, sizeData []int64, ...)
func AtgRandintOut(ptr *Ctensor, out Ctensor, high int64, sizeData []int64, sizeLen int)
func AtgRandn(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgRandnLike(ptr *Ctensor, self Ctensor)
func AtgRandnLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgRandnOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)
func AtgRandom(ptr *Ctensor, self Ctensor)
func AtgRandomFrom(ptr *Ctensor, self Ctensor, from int64, toVal int64, toNull int)
func AtgRandomFromOut(ptr *Ctensor, out Ctensor, self Ctensor, from int64, toVal int64, toNull int)
func AtgRandomFrom_(ptr *Ctensor, self Ctensor, from int64, toVal int64, toNull int)
func AtgRandomOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgRandomTo(ptr *Ctensor, self Ctensor, to int64)
func AtgRandomToOut(ptr *Ctensor, out Ctensor, self Ctensor, to int64)
func AtgRandomTo_(ptr *Ctensor, self Ctensor, to int64)
func AtgRandom_(ptr *Ctensor, self Ctensor)
func AtgRandperm(ptr *Ctensor, n int64, optionsKind int32, optionsDevice int32)
func AtgRandpermOut(ptr *Ctensor, out Ctensor, n int64)
func AtgRange(ptr *Ctensor, start Cscalar, end Cscalar, optionsKind int32, ...)
func AtgRangeOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar)
func AtgRangeOut_(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar)
func AtgRangeStep(ptr *Ctensor, start Cscalar, end Cscalar, optionsKind int32, ...)
func AtgRavel(ptr *Ctensor, self Ctensor)
func AtgReal(ptr *Ctensor, self Ctensor)
func AtgReciprocal(ptr *Ctensor, self Ctensor)
func AtgReciprocalOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgReciprocal_(ptr *Ctensor, self Ctensor)
func AtgReflectionPad1d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReflectionPad1dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, ...)
func AtgReflectionPad1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgReflectionPad1dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReflectionPad2d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReflectionPad2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, ...)
func AtgReflectionPad2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgReflectionPad2dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReflectionPad3d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReflectionPad3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, ...)
func AtgReflectionPad3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgReflectionPad3dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgRelu(ptr *Ctensor, self Ctensor)
func AtgRelu6(ptr *Ctensor, self Ctensor)
func AtgRelu6_(ptr *Ctensor, self Ctensor)
func AtgReluOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgRelu_(ptr *Ctensor, self Ctensor)
func AtgRemainder(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgRemainderScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgRemainderScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgRemainderScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgRemainderTensor(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgRemainderTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgRemainderTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgRemainder_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgRenorm(ptr *Ctensor, self Ctensor, p Cscalar, dim int64, maxnorm Cscalar)
func AtgRenormOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dim int64, maxnorm Cscalar)
func AtgRenorm_(ptr *Ctensor, self Ctensor, p Cscalar, dim int64, maxnorm Cscalar)
func AtgRepeat(ptr *Ctensor, self Ctensor, repeatsData []int64, repeatsLen int)
func AtgRepeatInterleave(ptr *Ctensor, repeats Ctensor, outputSizeVal int64, outputSizeNull int)
func AtgRepeatInterleaveSelfInt(ptr *Ctensor, self Ctensor, repeats int64, dimVal int64, dimNull int, ...)
func AtgRepeatInterleaveSelfTensor(ptr *Ctensor, self Ctensor, repeats Ctensor, dimVal int64, dimNull int, ...)
func AtgRepeatInterleaveTensorOut(ptr *Ctensor, out Ctensor, repeats Ctensor, outputSizeVal int64, ...)
func AtgRepeatOut(ptr *Ctensor, out Ctensor, self Ctensor, repeatsData []int64, repeatsLen int)
func AtgReplicationPad1d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReplicationPad1dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, ...)
func AtgReplicationPad1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgReplicationPad1dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReplicationPad2d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReplicationPad2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, ...)
func AtgReplicationPad2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgReplicationPad2dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReplicationPad3d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgReplicationPad3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, ...)
func AtgReplicationPad3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgReplicationPad3dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)
func AtgRequiresGrad_(ptr *Ctensor, self Ctensor, requiresGrad int32)
func AtgReshape(ptr *Ctensor, self Ctensor, shapeData []int64, shapeLen int)
func AtgReshapeAs(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgResize(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgResizeAs(ptr *Ctensor, self Ctensor, theTemplate Ctensor)
func AtgResizeAsOut(ptr *Ctensor, out Ctensor, self Ctensor, theTemplate Ctensor)
func AtgResizeAsSparse(ptr *Ctensor, self Ctensor, theTemplate Ctensor)
func AtgResizeAsSparseOut(ptr *Ctensor, out Ctensor, self Ctensor, theTemplate Ctensor)
func AtgResizeAsSparse_(ptr *Ctensor, self Ctensor, theTemplate Ctensor)
func AtgResizeAs_(ptr *Ctensor, self Ctensor, theTemplate Ctensor)
func AtgResizeOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgResize_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgResolveConj(ptr *Ctensor, self Ctensor)
func AtgResolveNeg(ptr *Ctensor, self Ctensor)
func AtgRetainsGrad(self Ctensor) bool
func AtgRnnRelu(ptr *Ctensor, input Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, ...)
func AtgRnnReluCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, ...)
func AtgRnnReluData(ptr *Ctensor, data Ctensor, batchSizes Ctensor, hx Ctensor, ...)
func AtgRnnTanh(ptr *Ctensor, input Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, ...)
func AtgRnnTanhCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, ...)
func AtgRnnTanhData(ptr *Ctensor, data Ctensor, batchSizes Ctensor, hx Ctensor, ...)
func AtgRoll(ptr *Ctensor, self Ctensor, shiftsData []int64, shiftsLen int, ...)
func AtgRollOut(ptr *Ctensor, out Ctensor, self Ctensor, shiftsData []int64, shiftsLen int, ...)
func AtgRot90(ptr *Ctensor, self Ctensor, k int64, dimsData []int64, dimsLen int)
func AtgRot90Out(ptr *Ctensor, out Ctensor, self Ctensor, k int64, dimsData []int64, ...)
func AtgRound(ptr *Ctensor, self Ctensor)
func AtgRoundDecimals(ptr *Ctensor, self Ctensor, decimals int64)
func AtgRoundDecimalsOut(ptr *Ctensor, out Ctensor, self Ctensor, decimals int64)
func AtgRoundDecimals_(ptr *Ctensor, self Ctensor, decimals int64)
func AtgRoundOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgRound_(ptr *Ctensor, self Ctensor)
func AtgRowIndices(ptr *Ctensor, self Ctensor)
func AtgRowIndicesCopy(ptr *Ctensor, self Ctensor)
func AtgRowIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgRowStack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgRowStackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgRrelu(ptr *Ctensor, self Ctensor, training int32)
func AtgRreluWithNoise(ptr *Ctensor, self Ctensor, noise Ctensor, training int32)
func AtgRreluWithNoiseBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, noise Ctensor, lower Cscalar, ...)
func AtgRreluWithNoiseBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor, noise Ctensor, ...)
func AtgRreluWithNoiseOut(ptr *Ctensor, out Ctensor, self Ctensor, noise Ctensor, training int32)
func AtgRreluWithNoise_(ptr *Ctensor, self Ctensor, noise Ctensor, training int32)
func AtgRrelu_(ptr *Ctensor, self Ctensor, training int32)
func AtgRsqrt(ptr *Ctensor, self Ctensor)
func AtgRsqrtOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgRsqrt_(ptr *Ctensor, self Ctensor)
func AtgRsub(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgRsubScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgRsubScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgRsubTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgScalarTensor(ptr *Ctensor, s Cscalar, optionsKind int32, optionsDevice int32)
func AtgScalarTensorOut(ptr *Ctensor, out Ctensor, s Cscalar)
func AtgScaledDotProductAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, ...)
func AtgScatter(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)
func AtgScatterAdd(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)
func AtgScatterAddOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)
func AtgScatterAdd_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)
func AtgScatterReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, ...)
func AtgScatterReduceOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, ...)
func AtgScatterReduce_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, ...)
func AtgScatterSrcOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)
func AtgScatterValue(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)
func AtgScatterValueOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, ...)
func AtgScatterValueReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar, ...)
func AtgScatterValueReduceOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, ...)
func AtgScatterValueReduce_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar, ...)
func AtgScatterValue_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)
func AtgScatter_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)
func AtgSearchsorted(ptr *Ctensor, sortedSequence Ctensor, self Ctensor, outInt32 int32, ...)
func AtgSearchsortedScalar(ptr *Ctensor, sortedSequence Ctensor, selfScalar Cscalar, outInt32 int32, ...)
func AtgSearchsortedScalarOut(ptr *Ctensor, out Ctensor, sortedSequence Ctensor, selfScalar Cscalar, ...)
func AtgSearchsortedTensorOut(ptr *Ctensor, out Ctensor, sortedSequence Ctensor, self Ctensor, ...)
func AtgSegmentReduce(ptr *Ctensor, data Ctensor, reduce string, lengths Ctensor, indices Ctensor, ...)
func AtgSegmentReduceOut(ptr *Ctensor, out Ctensor, data Ctensor, reduce string, lengths Ctensor, ...)
func AtgSelect(ptr *Ctensor, self Ctensor, dim int64, index int64)
func AtgSelectBackward(ptr *Ctensor, gradOutput Ctensor, inputSizesData []int64, inputSizesLen int, ...)
func AtgSelectBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, inputSizesData []int64, ...)
func AtgSelectCopy(ptr *Ctensor, self Ctensor, dim int64, index int64)
func AtgSelectCopyIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index int64)
func AtgSelectScatter(ptr *Ctensor, self Ctensor, src Ctensor, dim int64, index int64)
func AtgSelectScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, dim int64, index int64)
func AtgSelu(ptr *Ctensor, self Ctensor)
func AtgSelu_(ptr *Ctensor, self Ctensor)
func AtgSet(ptr *Ctensor, self Ctensor)
func AtgSetOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSetRequiresGrad(ptr *Ctensor, self Ctensor, r int32)
func AtgSetSourceTensor(ptr *Ctensor, self Ctensor, source Ctensor)
func AtgSetSourceTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, source Ctensor)
func AtgSetSourceTensorStorageOffset_(ptr *Ctensor, self Ctensor, source Ctensor, storageOffset int64, ...)
func AtgSetSourceTensor_(ptr *Ctensor, self Ctensor, source Ctensor)
func AtgSet_(ptr *Ctensor, self Ctensor)
func AtgSgn(ptr *Ctensor, self Ctensor)
func AtgSgnOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSgn_(ptr *Ctensor, self Ctensor)
func AtgSigmoid(ptr *Ctensor, self Ctensor)
func AtgSigmoidBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor)
func AtgSigmoidBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, output Ctensor)
func AtgSigmoidOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSigmoid_(ptr *Ctensor, self Ctensor)
func AtgSign(ptr *Ctensor, self Ctensor)
func AtgSignOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSign_(ptr *Ctensor, self Ctensor)
func AtgSignbit(ptr *Ctensor, self Ctensor)
func AtgSignbitOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSilu(ptr *Ctensor, self Ctensor)
func AtgSiluBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)
func AtgSiluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor)
func AtgSiluOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSilu_(ptr *Ctensor, self Ctensor)
func AtgSin(ptr *Ctensor, self Ctensor)
func AtgSinOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSin_(ptr *Ctensor, self Ctensor)
func AtgSinc(ptr *Ctensor, self Ctensor)
func AtgSincOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSinc_(ptr *Ctensor, self Ctensor)
func AtgSinh(ptr *Ctensor, self Ctensor)
func AtgSinhOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSinh_(ptr *Ctensor, self Ctensor)
func AtgSlice(ptr *Ctensor, self Ctensor, dim int64, startVal int64, startNull int, ...)
func AtgSliceBackward(ptr *Ctensor, gradOutput Ctensor, inputSizesData []int64, inputSizesLen int, ...)
func AtgSliceBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, inputSizesData []int64, ...)
func AtgSliceCopy(ptr *Ctensor, self Ctensor, dim int64, startVal int64, startNull int, ...)
func AtgSliceCopyTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, startVal int64, ...)
func AtgSliceScatter(ptr *Ctensor, self Ctensor, src Ctensor, dim int64, startVal int64, ...)
func AtgSliceScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, dim int64, ...)
func AtgSlogdet(ptr *Ctensor, self Ctensor)
func AtgSlogdetOut(ptr *Ctensor, sign Ctensor, logabsdet Ctensor, self Ctensor)
func AtgSlowConv3d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, ...)
func AtgSlowConv3dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, ...)
func AtgSlowConvDilated2d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, ...)
func AtgSlowConvDilated2dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, ...)
func AtgSlowConvDilated3d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, ...)
func AtgSlowConvDilated3dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, ...)
func AtgSlowConvTranspose2d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, ...)
func AtgSlowConvTranspose2dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, ...)
func AtgSlowConvTranspose3d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, ...)
func AtgSlowConvTranspose3dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, ...)
func AtgSmm(ptr *Ctensor, self Ctensor, mat2 Ctensor)
func AtgSmoothL1Loss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64, beta float64)
func AtgSmoothL1LossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, ...)
func AtgSmoothL1LossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgSmoothL1LossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64, ...)
func AtgSoftMarginLoss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64)
func AtgSoftMarginLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, ...)
func AtgSoftMarginLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgSoftMarginLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64)
func AtgSoftmax(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func AtgSoftmaxIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, dtype int32)
func AtgSoftplus(ptr *Ctensor, self Ctensor)
func AtgSoftplusBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, beta Cscalar, ...)
func AtgSoftplusBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgSoftplusOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSoftshrink(ptr *Ctensor, self Ctensor)
func AtgSoftshrinkBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, lambd Cscalar)
func AtgSoftshrinkBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgSoftshrinkOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSort(ptr *Ctensor, self Ctensor, dim int64, descending int32)
func AtgSortStable(ptr *Ctensor, self Ctensor, stable int32, dim int64, descending int32)
func AtgSortValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64, ...)
func AtgSortValuesStable(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, stable int32, ...)
func AtgSparseBscTensor(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, ...)
func AtgSparseBscTensorCcolRowValueSize(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, ...)
func AtgSparseBsrTensor(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, ...)
func AtgSparseBsrTensorCrowColValueSize(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, ...)
func AtgSparseCompressedTensor(ptr *Ctensor, compressedIndices Ctensor, plainIndices Ctensor, values Ctensor, ...)
func AtgSparseCompressedTensorCompPlainValueSize(ptr *Ctensor, compressedIndices Ctensor, plainIndices Ctensor, values Ctensor, ...)
func AtgSparseCooTensor(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgSparseCooTensorIndices(ptr *Ctensor, indices Ctensor, values Ctensor, optionsKind int32, ...)
func AtgSparseCooTensorIndicesSize(ptr *Ctensor, indices Ctensor, values Ctensor, sizeData []int64, sizeLen int, ...)
func AtgSparseCooTensorSizeOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)
func AtgSparseCscTensor(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, ...)
func AtgSparseCscTensorCcolRowValueSize(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, ...)
func AtgSparseCsrTensor(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, ...)
func AtgSparseCsrTensorCrowColValueSize(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, ...)
func AtgSparseDim(self Ctensor) int64
func AtgSparseMask(ptr *Ctensor, self Ctensor, mask Ctensor)
func AtgSparseMaskOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor)
func AtgSparseResize(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, ...)
func AtgSparseResizeAndClear(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, ...)
func AtgSparseResizeAndClearOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, ...)
func AtgSparseResizeAndClear_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, ...)
func AtgSparseResizeOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, ...)
func AtgSparseResize_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, ...)
func AtgSparseSampledAddmm(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)
func AtgSparseSampledAddmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)
func AtgSpecialAiryAi(ptr *Ctensor, x Ctensor)
func AtgSpecialAiryAiOut(ptr *Ctensor, out Ctensor, x Ctensor)
func AtgSpecialBesselJ0(ptr *Ctensor, self Ctensor)
func AtgSpecialBesselJ0Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialBesselJ1(ptr *Ctensor, self Ctensor)
func AtgSpecialBesselJ1Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialBesselY0(ptr *Ctensor, self Ctensor)
func AtgSpecialBesselY0Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialBesselY1(ptr *Ctensor, self Ctensor)
func AtgSpecialBesselY1Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialChebyshevPolynomialT(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialChebyshevPolynomialTNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialChebyshevPolynomialTNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialChebyshevPolynomialTOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialChebyshevPolynomialTXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialChebyshevPolynomialTXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialChebyshevPolynomialU(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialChebyshevPolynomialUNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialChebyshevPolynomialUNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialChebyshevPolynomialUOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialChebyshevPolynomialUXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialChebyshevPolynomialUXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialChebyshevPolynomialV(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialChebyshevPolynomialVNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialChebyshevPolynomialVNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialChebyshevPolynomialVOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialChebyshevPolynomialVXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialChebyshevPolynomialVXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialChebyshevPolynomialW(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialChebyshevPolynomialWNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialChebyshevPolynomialWNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialChebyshevPolynomialWOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialChebyshevPolynomialWXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialChebyshevPolynomialWXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialDigamma(ptr *Ctensor, self Ctensor)
func AtgSpecialDigammaOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialEntr(ptr *Ctensor, self Ctensor)
func AtgSpecialEntrOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialErf(ptr *Ctensor, self Ctensor)
func AtgSpecialErfOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialErfc(ptr *Ctensor, self Ctensor)
func AtgSpecialErfcOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialErfcx(ptr *Ctensor, self Ctensor)
func AtgSpecialErfcxOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialErfinv(ptr *Ctensor, self Ctensor)
func AtgSpecialErfinvOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialExp2(ptr *Ctensor, self Ctensor)
func AtgSpecialExp2Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialExpit(ptr *Ctensor, self Ctensor)
func AtgSpecialExpitOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialExpm1(ptr *Ctensor, self Ctensor)
func AtgSpecialExpm1Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialGammainc(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgSpecialGammaincOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgSpecialGammaincc(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgSpecialGammainccOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgSpecialGammaln(ptr *Ctensor, self Ctensor)
func AtgSpecialGammalnOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialHermitePolynomialH(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialHermitePolynomialHNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialHermitePolynomialHNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialHermitePolynomialHOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialHermitePolynomialHXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialHermitePolynomialHXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialHermitePolynomialHe(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialHermitePolynomialHeNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialHermitePolynomialHeNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialHermitePolynomialHeOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialHermitePolynomialHeXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialHermitePolynomialHeXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialI0(ptr *Ctensor, self Ctensor)
func AtgSpecialI0Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialI0e(ptr *Ctensor, self Ctensor)
func AtgSpecialI0eOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialI1(ptr *Ctensor, self Ctensor)
func AtgSpecialI1Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialI1e(ptr *Ctensor, self Ctensor)
func AtgSpecialI1eOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialLaguerrePolynomialL(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialLaguerrePolynomialLNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialLaguerrePolynomialLNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialLaguerrePolynomialLOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialLaguerrePolynomialLXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialLaguerrePolynomialLXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialLegendrePolynomialP(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialLegendrePolynomialPNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialLegendrePolynomialPNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialLegendrePolynomialPOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialLegendrePolynomialPXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialLegendrePolynomialPXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialLog1p(ptr *Ctensor, self Ctensor)
func AtgSpecialLog1pOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialLogNdtr(ptr *Ctensor, self Ctensor)
func AtgSpecialLogNdtrOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialLogSoftmax(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func AtgSpecialLogit(ptr *Ctensor, self Ctensor, epsVal float64, epsNull int)
func AtgSpecialLogitOut(ptr *Ctensor, out Ctensor, self Ctensor, epsVal float64, epsNull int)
func AtgSpecialLogsumexp(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)
func AtgSpecialLogsumexpOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgSpecialModifiedBesselI0(ptr *Ctensor, self Ctensor)
func AtgSpecialModifiedBesselI0Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialModifiedBesselI1(ptr *Ctensor, self Ctensor)
func AtgSpecialModifiedBesselI1Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialModifiedBesselK0(ptr *Ctensor, self Ctensor)
func AtgSpecialModifiedBesselK0Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialModifiedBesselK1(ptr *Ctensor, self Ctensor)
func AtgSpecialModifiedBesselK1Out(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialMultigammaln(ptr *Ctensor, self Ctensor, p int64)
func AtgSpecialMultigammalnOut(ptr *Ctensor, out Ctensor, self Ctensor, p int64)
func AtgSpecialNdtr(ptr *Ctensor, self Ctensor)
func AtgSpecialNdtrOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialNdtri(ptr *Ctensor, self Ctensor)
func AtgSpecialNdtriOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialPolygamma(ptr *Ctensor, n int64, self Ctensor)
func AtgSpecialPolygammaOut(ptr *Ctensor, out Ctensor, n int64, self Ctensor)
func AtgSpecialPsi(ptr *Ctensor, self Ctensor)
func AtgSpecialPsiOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialRound(ptr *Ctensor, self Ctensor, decimals int64)
func AtgSpecialRoundOut(ptr *Ctensor, out Ctensor, self Ctensor, decimals int64)
func AtgSpecialScaledModifiedBesselK0(ptr *Ctensor, x Ctensor)
func AtgSpecialScaledModifiedBesselK0Out(ptr *Ctensor, out Ctensor, x Ctensor)
func AtgSpecialScaledModifiedBesselK1(ptr *Ctensor, x Ctensor)
func AtgSpecialScaledModifiedBesselK1Out(ptr *Ctensor, out Ctensor, x Ctensor)
func AtgSpecialShiftedChebyshevPolynomialT(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialTNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialShiftedChebyshevPolynomialTNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialShiftedChebyshevPolynomialTOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialTXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialTXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialU(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialUNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialShiftedChebyshevPolynomialUNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialShiftedChebyshevPolynomialUOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialUXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialUXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialV(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialVNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialShiftedChebyshevPolynomialVNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialShiftedChebyshevPolynomialVOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialVXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialVXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialW(ptr *Ctensor, x Ctensor, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialWNScalar(ptr *Ctensor, x Ctensor, n Cscalar)
func AtgSpecialShiftedChebyshevPolynomialWNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)
func AtgSpecialShiftedChebyshevPolynomialWOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialWXScalar(ptr *Ctensor, x Cscalar, n Ctensor)
func AtgSpecialShiftedChebyshevPolynomialWXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)
func AtgSpecialSinc(ptr *Ctensor, self Ctensor)
func AtgSpecialSincOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSpecialSoftmax(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func AtgSpecialSphericalBesselJ0(ptr *Ctensor, x Ctensor)
func AtgSpecialSphericalBesselJ0Out(ptr *Ctensor, out Ctensor, x Ctensor)
func AtgSpecialXlog1py(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgSpecialXlog1pyOtherScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgSpecialXlog1pyOtherScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgSpecialXlog1pyOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgSpecialXlog1pySelfScalar(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgSpecialXlog1pySelfScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgSpecialXlogy(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgSpecialXlogyOtherScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgSpecialXlogyOtherScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgSpecialXlogyOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgSpecialXlogySelfScalar(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgSpecialXlogySelfScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgSpecialZeta(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgSpecialZetaOtherScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgSpecialZetaOtherScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgSpecialZetaOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgSpecialZetaSelfScalar(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgSpecialZetaSelfScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgSqrt(ptr *Ctensor, self Ctensor)
func AtgSqrtOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSqrt_(ptr *Ctensor, self Ctensor)
func AtgSquare(ptr *Ctensor, self Ctensor)
func AtgSquareOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSquare_(ptr *Ctensor, self Ctensor)
func AtgSqueeze(ptr *Ctensor, self Ctensor)
func AtgSqueezeCopy(ptr *Ctensor, self Ctensor)
func AtgSqueezeCopyDim(ptr *Ctensor, self Ctensor, dim int64)
func AtgSqueezeCopyDimOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)
func AtgSqueezeCopyDims(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)
func AtgSqueezeCopyDimsOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int)
func AtgSqueezeCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgSqueezeDim(ptr *Ctensor, self Ctensor, dim int64)
func AtgSqueezeDim_(ptr *Ctensor, self Ctensor, dim int64)
func AtgSqueezeDims(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)
func AtgSqueezeDims_(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)
func AtgSqueeze_(ptr *Ctensor, self Ctensor)
func AtgSspaddmm(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)
func AtgSspaddmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)
func AtgStack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func AtgStackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func AtgStd(ptr *Ctensor, self Ctensor, unbiased int32)
func AtgStdCorrection(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, ...)
func AtgStdCorrectionOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgStdDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, ...)
func AtgStdMean(ptr *Ctensor, self Ctensor, unbiased int32)
func AtgStdMeanCorrection(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, ...)
func AtgStdMeanCorrectionOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, dimData []int64, ...)
func AtgStdMeanDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, ...)
func AtgStdOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgStft(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, ...)
func AtgStftCenter(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, ...)
func AtgSub(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgSubOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgSubScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgSubScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgSubScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgSub_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgSubtract(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgSubtractOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgSubtractScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgSubtractScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgSubtract_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgSum(ptr *Ctensor, self Ctensor, dtype int32)
func AtgSumDimIntlist(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, ...)
func AtgSumIntlistOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgSumOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32)
func AtgSumToSize(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgSvd(ptr *Ctensor, self Ctensor, some int32, computeUv int32)
func AtgSvdU(ptr *Ctensor, u Ctensor, s Ctensor, v Ctensor, self Ctensor, some int32, ...)
func AtgSwapaxes(ptr *Ctensor, self Ctensor, axis0 int64, axis1 int64)
func AtgSwapaxes_(ptr *Ctensor, self Ctensor, axis0 int64, axis1 int64)
func AtgSwapdims(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)
func AtgSwapdims_(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)
func AtgT(ptr *Ctensor, self Ctensor)
func AtgTCopy(ptr *Ctensor, self Ctensor)
func AtgTCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgT_(ptr *Ctensor, self Ctensor)
func AtgTake(ptr *Ctensor, self Ctensor, index Ctensor)
func AtgTakeAlongDim(ptr *Ctensor, self Ctensor, indices Ctensor, dimVal int64, dimNull int)
func AtgTakeAlongDimOut(ptr *Ctensor, out Ctensor, self Ctensor, indices Ctensor, dimVal int64, ...)
func AtgTakeOut(ptr *Ctensor, out Ctensor, self Ctensor, index Ctensor)
func AtgTan(ptr *Ctensor, self Ctensor)
func AtgTanOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgTan_(ptr *Ctensor, self Ctensor)
func AtgTanh(ptr *Ctensor, self Ctensor)
func AtgTanhBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor)
func AtgTanhBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, output Ctensor)
func AtgTanhOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgTanh_(ptr *Ctensor, self Ctensor)
func AtgTensordot(ptr *Ctensor, self Ctensor, other Ctensor, dimsSelfData []int64, ...)
func AtgTensordotOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, dimsSelfData []int64, ...)
func AtgThreshold(ptr *Ctensor, self Ctensor, threshold Cscalar, value Cscalar)
func AtgThresholdBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, threshold Cscalar)
func AtgThresholdBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, ...)
func AtgThresholdOut(ptr *Ctensor, out Ctensor, self Ctensor, threshold Cscalar, value Cscalar)
func AtgThreshold_(ptr *Ctensor, self Ctensor, threshold Cscalar, value Cscalar)
func AtgTile(ptr *Ctensor, self Ctensor, dimsData []int64, dimsLen int)
func AtgTo(ptr *Ctensor, self Ctensor, device int32)
func AtgToDense(ptr *Ctensor, self Ctensor, dtype int32, maskedGrad int32)
func AtgToDenseBackward(ptr *Ctensor, grad Ctensor, input Ctensor, maskedGrad int32)
func AtgToDevice(ptr *Ctensor, self Ctensor, device int32, dtype int32, nonBlocking int32, ...)
func AtgToDtype(ptr *Ctensor, self Ctensor, dtype int32, nonBlocking int32, copy int32)
func AtgToDtypeLayout(ptr *Ctensor, self Ctensor, optionsKind int32, optionsDevice int32, ...)
func AtgToMkldnn(ptr *Ctensor, self Ctensor, dtype int32)
func AtgToMkldnnBackward(ptr *Ctensor, grad Ctensor, input Ctensor)
func AtgToMkldnnOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32)
func AtgToOther(ptr *Ctensor, self Ctensor, other Ctensor, nonBlocking int32, copy int32)
func AtgToPaddedTensor(ptr *Ctensor, self Ctensor, padding float64, outputSizeData []int64, ...)
func AtgToPaddedTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, padding float64, ...)
func AtgToSparse(ptr *Ctensor, self Ctensor, layout int8, blocksizeData []int64, ...)
func AtgToSparseBsc(ptr *Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, ...)
func AtgToSparseBsr(ptr *Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, ...)
func AtgToSparseCsc(ptr *Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)
func AtgToSparseCsr(ptr *Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)
func AtgToSparseSparseDim(ptr *Ctensor, self Ctensor, sparseDim int64)
func AtgTopk(ptr *Ctensor, self Ctensor, k int64, dim int64, largest int32, sorted int32)
func AtgTopkValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, k int64, ...)
func AtgTotype(ptr *Ctensor, self Ctensor, scalarType int32)
func AtgTrace(ptr *Ctensor, self Ctensor)
func AtgTraceBackward(ptr *Ctensor, grad Ctensor, sizesData []int64, sizesLen int)
func AtgTraceOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgTranspose(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)
func AtgTransposeCopy(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)
func AtgTransposeCopyIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim0 int64, dim1 int64)
func AtgTranspose_(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)
func AtgTrapezoid(ptr *Ctensor, y Ctensor, dim int64)
func AtgTrapezoidX(ptr *Ctensor, y Ctensor, x Ctensor, dim int64)
func AtgTrapz(ptr *Ctensor, y Ctensor, x Ctensor, dim int64)
func AtgTrapzDx(ptr *Ctensor, y Ctensor, dx float64, dim int64)
func AtgTriangularSolve(ptr *Ctensor, self Ctensor, a Ctensor, upper int32, transpose int32, ...)
func AtgTriangularSolveX(ptr *Ctensor, x Ctensor, m Ctensor, self Ctensor, a Ctensor, upper int32, ...)
func AtgTril(ptr *Ctensor, self Ctensor, diagonal int64)
func AtgTrilIndices(ptr *Ctensor, row int64, col int64, offset int64, optionsKind int32, ...)
func AtgTrilIndicesOut(ptr *Ctensor, out Ctensor, row int64, col int64, offset int64)
func AtgTrilOut(ptr *Ctensor, out Ctensor, self Ctensor, diagonal int64)
func AtgTril_(ptr *Ctensor, self Ctensor, diagonal int64)
func AtgTripletMarginLoss(ptr *Ctensor, anchor Ctensor, positive Ctensor, negative Ctensor, ...)
func AtgTriu(ptr *Ctensor, self Ctensor, diagonal int64)
func AtgTriuIndices(ptr *Ctensor, row int64, col int64, offset int64, optionsKind int32, ...)
func AtgTriuIndicesOut(ptr *Ctensor, out Ctensor, row int64, col int64, offset int64)
func AtgTriuOut(ptr *Ctensor, out Ctensor, self Ctensor, diagonal int64)
func AtgTriu_(ptr *Ctensor, self Ctensor, diagonal int64)
func AtgTrueDivide(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgTrueDivideOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgTrueDivideScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgTrueDivideScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgTrueDivide_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgTrunc(ptr *Ctensor, self Ctensor)
func AtgTruncOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgTrunc_(ptr *Ctensor, self Ctensor)
func AtgTypeAs(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgUnflatten(ptr *Ctensor, self Ctensor, dim int64, sizesData []int64, sizesLen int)
func AtgUnfold(ptr *Ctensor, self Ctensor, dimension int64, size int64, step int64)
func AtgUnfoldBackward(ptr *Ctensor, gradIn Ctensor, inputSizesData []int64, inputSizesLen int, ...)
func AtgUnfoldBackwardOut(ptr *Ctensor, out Ctensor, gradIn Ctensor, inputSizesData []int64, ...)
func AtgUnfoldCopy(ptr *Ctensor, self Ctensor, dimension int64, size int64, step int64)
func AtgUnfoldCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dimension int64, size int64, ...)
func AtgUniform(ptr *Ctensor, self Ctensor, from float64, to float64)
func AtgUniformOut(ptr *Ctensor, out Ctensor, self Ctensor, from float64, to float64)
func AtgUniform_(ptr *Ctensor, self Ctensor, from float64, to float64)
func AtgUniqueConsecutive(ptr *Ctensor, self Ctensor, returnInverse int32, returnCounts int32, ...)
func AtgUniqueConsecutiveOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, self Ctensor, ...)
func AtgUniqueDim(ptr *Ctensor, self Ctensor, dim int64, sorted int32, returnInverse int32, ...)
func AtgUniqueDimConsecutive(ptr *Ctensor, self Ctensor, dim int64, returnInverse int32, returnCounts int32)
func AtgUniqueDimConsecutiveOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, self Ctensor, ...)
func AtgUniqueDimOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, self Ctensor, ...)
func AtgUnsqueeze(ptr *Ctensor, self Ctensor, dim int64)
func AtgUnsqueezeCopy(ptr *Ctensor, self Ctensor, dim int64)
func AtgUnsqueezeCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)
func AtgUnsqueeze_(ptr *Ctensor, self Ctensor, dim int64)
func AtgUpsampleBicubic2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleBicubic2dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleBicubic2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func AtgUpsampleBicubic2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgUpsampleBicubic2dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleBilinear2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleBilinear2dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleBilinear2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func AtgUpsampleBilinear2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgUpsampleBilinear2dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleLinear1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleLinear1dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleLinear1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func AtgUpsampleLinear1dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgUpsampleLinear1dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleNearest1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleNearest1dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleNearest1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func AtgUpsampleNearest1dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgUpsampleNearest1dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleNearest2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleNearest2dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleNearest2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func AtgUpsampleNearest2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgUpsampleNearest2dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleNearest3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleNearest3dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleNearest3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func AtgUpsampleNearest3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgUpsampleNearest3dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleTrilinear3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleTrilinear3dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgUpsampleTrilinear3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func AtgUpsampleTrilinear3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func AtgUpsampleTrilinear3dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func AtgValueSelectingReductionBackward(ptr *Ctensor, grad Ctensor, dim int64, indices Ctensor, sizesData []int64, ...)
func AtgValues(ptr *Ctensor, self Ctensor)
func AtgValuesCopy(ptr *Ctensor, self Ctensor)
func AtgValuesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgVander(ptr *Ctensor, x Ctensor, nVal int64, nNull int, increasing int32)
func AtgVar(ptr *Ctensor, self Ctensor, unbiased int32)
func AtgVarCorrection(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, ...)
func AtgVarCorrectionOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgVarDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, ...)
func AtgVarMean(ptr *Ctensor, self Ctensor, unbiased int32)
func AtgVarMeanCorrection(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, ...)
func AtgVarMeanCorrectionOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, dimData []int64, ...)
func AtgVarMeanDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, ...)
func AtgVarOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func AtgVdot(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgVdotOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgView(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgViewAs(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgViewAsComplex(ptr *Ctensor, self Ctensor)
func AtgViewAsComplexCopy(ptr *Ctensor, self Ctensor)
func AtgViewAsComplexCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgViewAsReal(ptr *Ctensor, self Ctensor)
func AtgViewAsRealCopy(ptr *Ctensor, self Ctensor)
func AtgViewAsRealCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgViewCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgViewCopyDtype(ptr *Ctensor, self Ctensor, dtype int32)
func AtgViewCopyDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32)
func AtgViewCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func AtgViewDtype(ptr *Ctensor, self Ctensor, dtype int32)
func AtgVstack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgVstackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)
func AtgWhereScalar(ptr *Ctensor, condition Ctensor, selfScalar Cscalar, other Cscalar)
func AtgWhereScalarother(ptr *Ctensor, condition Ctensor, self Ctensor, other Cscalar)
func AtgWhereScalarself(ptr *Ctensor, condition Ctensor, selfScalar Cscalar, other Ctensor)
func AtgWhereSelf(ptr *Ctensor, condition Ctensor, self Ctensor, other Ctensor)
func AtgWhereSelfOut(ptr *Ctensor, out Ctensor, condition Ctensor, self Ctensor, other Ctensor)
func AtgXlogy(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgXlogyOutscalarOther(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func AtgXlogyOutscalarSelf(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)
func AtgXlogyOuttensor(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func AtgXlogyScalarOther(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgXlogyScalarOther_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtgXlogyScalarSelf(ptr *Ctensor, selfScalar Cscalar, other Ctensor)
func AtgXlogy_(ptr *Ctensor, self Ctensor, other Ctensor)
func AtgZero(ptr *Ctensor, self Ctensor)
func AtgZeroOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgZero_(ptr *Ctensor, self Ctensor)
func AtgZeros(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func AtgZerosLike(ptr *Ctensor, self Ctensor)
func AtgZerosLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)
func AtgZerosOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)
func Atg_AdaptiveAvgPool2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)
func Atg_AdaptiveAvgPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)
func Atg_AdaptiveAvgPool2dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor)
func Atg_AdaptiveAvgPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func Atg_AdaptiveAvgPool3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)
func Atg_AdaptiveAvgPool3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)
func Atg_AdaptiveAvgPool3dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor)
func Atg_AdaptiveAvgPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func Atg_AddBatchDim(ptr *Ctensor, self Ctensor, batchDim int64, level int64)
func Atg_AddRelu(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg_AddReluOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func Atg_AddReluScalar(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg_AddReluScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func Atg_AddReluScalar_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg_AddRelu_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg_AddmmActivation(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor, useGelu int32)
func Atg_AddmmActivationOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor, ...)
func Atg_Aminmax(ptr *Ctensor, self Ctensor)
func Atg_AminmaxDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)
func Atg_AminmaxDimOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, dim int64, ...)
func Atg_AminmaxOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor)
func Atg_AmpUpdateScale(ptr *Ctensor, self Ctensor, growthTracker Ctensor, foundInf Ctensor, ...)
func Atg_AmpUpdateScaleOut(ptr *Ctensor, out Ctensor, self Ctensor, growthTracker Ctensor, ...)
func Atg_AmpUpdateScale_(ptr *Ctensor, self Ctensor, growthTracker Ctensor, foundInf Ctensor, ...)
func Atg_AutocastToFullPrecision(ptr *Ctensor, self Ctensor, cudaEnabled int32, cpuEnabled int32)
func Atg_AutocastToReducedPrecision(ptr *Ctensor, self Ctensor, cudaEnabled int32, cpuEnabled int32, ...)
func Atg_CastByte(ptr *Ctensor, self Ctensor, nonBlocking int32)
func Atg_CastChar(ptr *Ctensor, self Ctensor, nonBlocking int32)
func Atg_CastDouble(ptr *Ctensor, self Ctensor, nonBlocking int32)
func Atg_CastFloat(ptr *Ctensor, self Ctensor, nonBlocking int32)
func Atg_CastHalf(ptr *Ctensor, self Ctensor, nonBlocking int32)
func Atg_CastInt(ptr *Ctensor, self Ctensor, nonBlocking int32)
func Atg_CastLong(ptr *Ctensor, self Ctensor, nonBlocking int32)
func Atg_CastShort(ptr *Ctensor, self Ctensor, nonBlocking int32)
func Atg_CdistBackward(ptr *Ctensor, grad Ctensor, x1 Ctensor, x2 Ctensor, p float64, cdist Ctensor)
func Atg_CdistBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, x1 Ctensor, x2 Ctensor, p float64, ...)
func Atg_CholeskySolveHelper(ptr *Ctensor, self Ctensor, a Ctensor, upper int32)
func Atg_CholeskySolveHelperOut(ptr *Ctensor, out Ctensor, self Ctensor, a Ctensor, upper int32)
func Atg_Coalesce(ptr *Ctensor, self Ctensor)
func Atg_CoalesceOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_Coalesced(ptr *Ctensor, self Ctensor, coalesced int32)
func Atg_CoalescedOut(ptr *Ctensor, out Ctensor, self Ctensor, coalesced int32)
func Atg_Coalesced_(ptr *Ctensor, self Ctensor, coalesced int32)
func Atg_ComputeLinearCombination(ptr *Ctensor, input Ctensor, coefficients Ctensor)
func Atg_ComputeLinearCombinationOut(ptr *Ctensor, out Ctensor, input Ctensor, coefficients Ctensor)
func Atg_Conj(ptr *Ctensor, self Ctensor)
func Atg_ConjCopy(ptr *Ctensor, self Ctensor)
func Atg_ConjCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_ConjPhysical(ptr *Ctensor, self Ctensor)
func Atg_ConjPhysicalOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_ConvDepthwise2d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, ...)
func Atg_ConvDepthwise2dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, ...)
func Atg_ConvertIndicesFromCooToCsr(ptr *Ctensor, self Ctensor, size int64, outInt32 int32)
func Atg_ConvertIndicesFromCooToCsrOut(ptr *Ctensor, out Ctensor, self Ctensor, size int64, outInt32 int32)
func Atg_ConvertIndicesFromCsrToCoo(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, outInt32 int32, ...)
func Atg_ConvertIndicesFromCsrToCooOut(ptr *Ctensor, out Ctensor, crowIndices Ctensor, colIndices Ctensor, ...)
func Atg_Convolution(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func Atg_ConvolutionDeprecated(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func Atg_ConvolutionMode(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, ...)
func Atg_ConvolutionOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, ...)
func Atg_CopyFrom(ptr *Ctensor, self Ctensor, dst Ctensor, nonBlocking int32)
func Atg_CopyFromAndResize(ptr *Ctensor, self Ctensor, dst Ctensor)
func Atg_CopyFromAndResizeOut(ptr *Ctensor, out Ctensor, self Ctensor, dst Ctensor)
func Atg_CopyFromOut(ptr *Ctensor, out Ctensor, self Ctensor, dst Ctensor, nonBlocking int32)
func Atg_CsltCompress(ptr *Ctensor, input Ctensor)
func Atg_CsltSparseMm(ptr *Ctensor, compressedA Ctensor, denseB Ctensor, bias Ctensor, ...)
func Atg_CtcLoss(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, ...)
func Atg_CtcLossBackward(ptr *Ctensor, grad Ctensor, logProbs Ctensor, targets Ctensor, ...)
func Atg_CtcLossBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, logProbs Ctensor, targets Ctensor, ...)
func Atg_CtcLossBackwardTensor(ptr *Ctensor, grad Ctensor, logProbs Ctensor, targets Ctensor, ...)
func Atg_CtcLossOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, logProbs Ctensor, targets Ctensor, ...)
func Atg_CtcLossTensor(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengths Ctensor, ...)
func Atg_CtcLossTensorOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, logProbs Ctensor, targets Ctensor, ...)
func Atg_CudnnCtcLoss(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, ...)
func Atg_CudnnCtcLossOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, logProbs Ctensor, targets Ctensor, ...)
func Atg_CudnnCtcLossTensor(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengths Ctensor, ...)
func Atg_CudnnInitDropoutState(ptr *Ctensor, dropout float64, train int32, dropoutSeed int64, ...)
func Atg_CudnnInitDropoutStateOut(ptr *Ctensor, out Ctensor, dropout float64, train int32, dropoutSeed int64)
func Atg_CudnnRnn(ptr *Ctensor, input Ctensor, weightData []Ctensor, weightLen int, ...)
func Atg_CudnnRnnFlattenWeight(ptr *Ctensor, weightArrData []Ctensor, weightArrLen int, weightStride0 int64, ...)
func Atg_CudnnRnnFlattenWeightOut(ptr *Ctensor, out Ctensor, weightArrData []Ctensor, weightArrLen int, ...)
func Atg_CudnnRnnOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, ...)
func Atg_DebugHasInternalOverlap(self Ctensor) int64
func Atg_DimArange(ptr *Ctensor, like Ctensor, dim int64)
func Atg_Dimi(self Ctensor) int64
func Atg_Dimv(self Ctensor) int64
func Atg_DirichletGrad(ptr *Ctensor, x Ctensor, alpha Ctensor, total Ctensor)
func Atg_DirichletGradOut(ptr *Ctensor, out Ctensor, x Ctensor, alpha Ctensor, total Ctensor)
func Atg_EfficientAttentionBackward(ptr *Ctensor, gradOut_ Ctensor, query Ctensor, key Ctensor, value Ctensor, ...)
func Atg_Efficientzerotensor(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func Atg_EfficientzerotensorOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)
func Atg_EmbeddingBag(ptr *Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, ...)
func Atg_EmbeddingBagBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, offsets Ctensor, ...)
func Atg_EmbeddingBagDenseBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, offset2bag Ctensor, ...)
func Atg_EmbeddingBagDenseBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, indices Ctensor, offset2bag Ctensor, ...)
func Atg_EmbeddingBagForwardOnly(ptr *Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, ...)
func Atg_EmbeddingBagForwardOnlyOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, ...)
func Atg_EmbeddingBagOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, ...)
func Atg_EmbeddingBagPerSampleWeightsBackward(ptr *Ctensor, grad Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, ...)
func Atg_EmbeddingBagPerSampleWeightsBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, weight Ctensor, indices Ctensor, ...)
func Atg_EmbeddingBagSparseBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, offsets Ctensor, ...)
func Atg_EmptyAffineQuantized(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, ...)
func Atg_EmptyAffineQuantizedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, scale float64, ...)
func Atg_EmptyPerChannelAffineQuantized(ptr *Ctensor, sizeData []int64, sizeLen int, scales Ctensor, ...)
func Atg_EmptyPerChannelAffineQuantizedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, scales Ctensor, ...)
func Atg_EuclideanDist(ptr *Ctensor, x1 Ctensor, x2 Ctensor)
func Atg_EuclideanDistOut(ptr *Ctensor, out Ctensor, x1 Ctensor, x2 Ctensor)
func Atg_FakeQuantizeLearnablePerChannelAffine(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64, ...)
func Atg_FakeQuantizeLearnablePerChannelAffineBackward(ptr *Ctensor, grad Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, ...)
func Atg_FakeQuantizeLearnablePerChannelAffineOut(ptr *Ctensor, out Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, ...)
func Atg_FakeQuantizeLearnablePerTensorAffine(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, quantMin int64, ...)
func Atg_FakeQuantizeLearnablePerTensorAffineBackward(ptr *Ctensor, grad Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, ...)
func Atg_FakeQuantizeLearnablePerTensorAffineOut(ptr *Ctensor, out Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, ...)
func Atg_FakeQuantizePerTensorAffineCachemaskTensorQparams(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, ...)
func Atg_FakeQuantizePerTensorAffineCachemaskTensorQparamsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, scale Ctensor, ...)
func Atg_FftC2c(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, normalization int64, ...)
func Atg_FftC2cOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func Atg_FftC2r(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, normalization int64, ...)
func Atg_FftC2rOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func Atg_FftR2c(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, normalization int64, ...)
func Atg_FftR2cOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func Atg_FillMemEffDropoutMask_(ptr *Ctensor, self Ctensor, dropoutP float64, seed int64, offset int64)
func Atg_FlashAttentionBackward(ptr *Ctensor, gradOut Ctensor, query Ctensor, key Ctensor, value Ctensor, ...)
func Atg_Foobar(ptr *Ctensor, self Ctensor, arg1 int32, arg2 int32, arg3 int32)
func Atg_FoobarOut(ptr *Ctensor, out Ctensor, self Ctensor, arg1 int32, arg2 int32, arg3 int32)
func Atg_FunctionalAssertAsync(ptr *Ctensor, self Ctensor, assertMsg string, depToken Ctensor)
func Atg_FunctionalSymConstrainRange(ptr *Ctensor, size Cscalar, minVal int64, minNull int, maxVal int64, ...)
func Atg_FunctionalSymConstrainRangeForSize(ptr *Ctensor, size Cscalar, minVal int64, minNull int, maxVal int64, ...)
func Atg_FusedDropout(ptr *Ctensor, self Ctensor, p float64)
func Atg_FusedDropoutOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, p float64)
func Atg_FusedMovingAvgObsFqHelper(ptr *Ctensor, self Ctensor, observerOn Ctensor, fakeQuantOn Ctensor, ...)
func Atg_FusedMovingAvgObsFqHelperFunctional(ptr *Ctensor, self Ctensor, observerOn Ctensor, fakeQuantOn Ctensor, ...)
func Atg_FusedMovingAvgObsFqHelperOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, observerOn Ctensor, ...)
func Atg_FusedSdpChoice(query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, ...) int64
func Atg_FwPrimal(ptr *Ctensor, self Ctensor, level int64)
func Atg_FwPrimalCopy(ptr *Ctensor, self Ctensor, level int64)
func Atg_FwPrimalCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, level int64)
func Atg_GatherSparseBackward(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, grad Ctensor)
func Atg_GridSampler2dCpuFallback(ptr *Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, ...)
func Atg_GridSampler2dCpuFallbackBackward(ptr *Ctensor, gradOutput Ctensor, input Ctensor, grid Ctensor, ...)
func Atg_GridSampler2dCpuFallbackOut(ptr *Ctensor, out Ctensor, input Ctensor, grid Ctensor, ...)
func Atg_HasCompatibleShallowCopyType(self Ctensor, from Ctensor) bool
func Atg_HasSameStorageNumel(self Ctensor, other Ctensor) bool
func Atg_HistogramddFromBinCts(ptr *Ctensor, out Ctensor, self Ctensor, binsData []int64, binsLen int, ...)
func Atg_HistogramddFromBinTensors(ptr *Ctensor, self Ctensor, binsData []Ctensor, binsLen int, weight Ctensor, ...)
func Atg_HistogramddFromBinTensorsOut(ptr *Ctensor, out Ctensor, self Ctensor, binsData []Ctensor, binsLen int, ...)
func Atg_IndexPutImpl(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, ...)
func Atg_IndexPutImplOut(ptr *Ctensor, out Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, ...)
func Atg_IndexPutImpl_(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, ...)
func Atg_Indices(ptr *Ctensor, self Ctensor)
func Atg_IndicesCopy(ptr *Ctensor, self Ctensor)
func Atg_IndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_IntMm(ptr *Ctensor, self Ctensor, mat2 Ctensor)
func Atg_IntMmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor)
func Atg_IsAllTrue(ptr *Ctensor, self Ctensor)
func Atg_IsAnyTrue(ptr *Ctensor, self Ctensor)
func Atg_IsZerotensor(self Ctensor) bool
func Atg_LinalgDet(ptr *Ctensor, a Ctensor)
func Atg_LinalgDetResult(ptr *Ctensor, result Ctensor, lU Ctensor, pivots Ctensor, a Ctensor)
func Atg_LinalgEigh(ptr *Ctensor, a Ctensor, uPLO string, computeV int32)
func Atg_LinalgEighEigenvalues(ptr *Ctensor, eigenvalues Ctensor, eigenvectors Ctensor, a Ctensor, ...)
func Atg_LinalgSlogdet(ptr *Ctensor, a Ctensor)
func Atg_LinalgSlogdetSign(ptr *Ctensor, sign Ctensor, logabsdet Ctensor, lU Ctensor, pivots Ctensor, ...)
func Atg_LinalgSolveEx(ptr *Ctensor, a Ctensor, b Ctensor, left int32, checkErrors int32)
func Atg_LinalgSolveExResult(ptr *Ctensor, result Ctensor, lU Ctensor, pivots Ctensor, info Ctensor, ...)
func Atg_LinalgSvd(ptr *Ctensor, a Ctensor, fullMatrices int32, computeUv int32, driver string)
func Atg_LinalgSvdU(ptr *Ctensor, u Ctensor, s Ctensor, vh Ctensor, a Ctensor, fullMatrices int32, ...)
func Atg_LogSoftmax(ptr *Ctensor, self Ctensor, dim int64, halfToFloat int32)
func Atg_LogSoftmaxBackwardData(ptr *Ctensor, gradOutput Ctensor, output Ctensor, dim int64, inputDtype int32)
func Atg_LogSoftmaxBackwardDataOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, dim int64, ...)
func Atg_LogSoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, halfToFloat int32)
func Atg_Logcumsumexp(ptr *Ctensor, self Ctensor, dim int64)
func Atg_LogcumsumexpOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)
func Atg_LstmMps(ptr *Ctensor, input Ctensor, hxData []Ctensor, hxLen int, paramsData []Ctensor, ...)
func Atg_LstmMpsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, ...)
func Atg_LuWithInfo(ptr *Ctensor, self Ctensor, pivot int32, checkErrors int32)
func Atg_MakeDepToken(ptr *Ctensor, optionsKind int32, optionsDevice int32)
func Atg_MakeDual(ptr *Ctensor, primal Ctensor, tangent Ctensor, level int64)
func Atg_MakeDualCopy(ptr *Ctensor, primal Ctensor, tangent Ctensor, level int64)
func Atg_MakeDualCopyOut(ptr *Ctensor, out Ctensor, primal Ctensor, tangent Ctensor, level int64)
func Atg_MakePerChannelQuantizedTensor(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64)
func Atg_MakePerChannelQuantizedTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, ...)
func Atg_MakePerTensorQuantizedTensor(ptr *Ctensor, self Ctensor, scale float64, zeroPoint int64)
func Atg_MakePerTensorQuantizedTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, scale float64, zeroPoint int64)
func Atg_MaskedScale(ptr *Ctensor, self Ctensor, mask Ctensor, scale float64)
func Atg_MaskedScaleOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, scale float64)
func Atg_MaskedSoftmax(ptr *Ctensor, self Ctensor, mask Ctensor, dimVal int64, dimNull int, ...)
func Atg_MaskedSoftmaxBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor, mask Ctensor, dimVal int64, ...)
func Atg_MaskedSoftmaxBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, mask Ctensor, ...)
func Atg_MaskedSoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, dimVal int64, ...)
func Atg_MkldnnReshape(ptr *Ctensor, self Ctensor, shapeData []int64, shapeLen int)
func Atg_MkldnnReshapeOut(ptr *Ctensor, out Ctensor, self Ctensor, shapeData []int64, shapeLen int)
func Atg_MkldnnTranspose(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)
func Atg_MkldnnTransposeOut(ptr *Ctensor, out Ctensor, self Ctensor, dim0 int64, dim1 int64)
func Atg_MkldnnTranspose_(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)
func Atg_MpsConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, ...)
func Atg_MpsConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, ...)
func Atg_MpsConvolutionTranspose(ptr *Ctensor, self Ctensor, weight Ctensor, paddingData []int64, ...)
func Atg_MpsConvolutionTransposeOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, paddingData []int64, ...)
func Atg_NativeBatchNormLegit(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, ...)
func Atg_NativeBatchNormLegitFunctional(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, ...)
func Atg_NativeBatchNormLegitNoStats(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, training int32, ...)
func Atg_NativeBatchNormLegitNoStatsOut(ptr *Ctensor, out Ctensor, saveMean Ctensor, saveInvstd Ctensor, input Ctensor, ...)
func Atg_NativeBatchNormLegitNoTraining(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, ...)
func Atg_NativeBatchNormLegitNoTrainingOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, ...)
func Atg_NativeBatchNormLegitOut(ptr *Ctensor, out Ctensor, saveMean Ctensor, saveInvstd Ctensor, input Ctensor, ...)
func Atg_NativeMultiHeadAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, embedDim int64, ...)
func Atg_NativeMultiHeadAttentionOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, query Ctensor, key Ctensor, ...)
func Atg_NegView(ptr *Ctensor, self Ctensor)
func Atg_NegViewCopy(ptr *Ctensor, self Ctensor)
func Atg_NegViewCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_NestedFromPadded(ptr *Ctensor, padded Ctensor, cpuNestedShapeExample Ctensor, ...)
func Atg_NestedFromPaddedAndNestedExample(ptr *Ctensor, padded Ctensor, ntExample Ctensor)
func Atg_NestedFromPaddedAndNestedExampleOut(ptr *Ctensor, out Ctensor, padded Ctensor, ntExample Ctensor)
func Atg_NestedFromPaddedOut(ptr *Ctensor, out Ctensor, padded Ctensor, cpuNestedShapeExample Ctensor, ...)
func Atg_NestedSelectBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, dim int64, index int64)
func Atg_NestedSumBackward(ptr *Ctensor, grad Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func Atg_NestedViewFromBuffer(ptr *Ctensor, self Ctensor, nestedSize Ctensor, nestedStrides Ctensor, ...)
func Atg_NestedViewFromBufferCopy(ptr *Ctensor, self Ctensor, nestedSize Ctensor, nestedStrides Ctensor, ...)
func Atg_NestedViewFromBufferCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, nestedSize Ctensor, ...)
func Atg_NewZerosWithSameFeatureMeta(ptr *Ctensor, self Ctensor, other Ctensor, selfNumBatchDims int64)
func Atg_NewZerosWithSameFeatureMetaOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, selfNumBatchDims int64)
func Atg_NnpackAvailable() bool
func Atg_NnpackSpatialConvolution(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, ...)
func Atg_NnpackSpatialConvolutionOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, ...)
func Atg_Nnz(self Ctensor) int64
func Atg_PackPaddedSequence(ptr *Ctensor, input Ctensor, lengths Ctensor, batchFirst int32)
func Atg_PackPaddedSequenceBackward(ptr *Ctensor, grad Ctensor, inputSizeData []int64, inputSizeLen int, ...)
func Atg_PackPaddedSequenceOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, lengths Ctensor, ...)
func Atg_PadCircular(ptr *Ctensor, self Ctensor, padData []int64, padLen int)
func Atg_PadEnum(ptr *Ctensor, self Ctensor, padData []int64, padLen int, mode int64, ...)
func Atg_PadPackedSequence(ptr *Ctensor, data Ctensor, batchSizes Ctensor, batchFirst int32, ...)
func Atg_PdistBackward(ptr *Ctensor, grad Ctensor, self Ctensor, p float64, pdist Ctensor)
func Atg_PdistBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, self Ctensor, p float64, ...)
func Atg_PinMemory(ptr *Ctensor, self Ctensor, device int32)
func Atg_PinMemoryOut(ptr *Ctensor, out Ctensor, self Ctensor, device int32)
func Atg_PreluKernel(ptr *Ctensor, self Ctensor, weight Ctensor)
func Atg_PreluKernelBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, weight Ctensor)
func Atg_RemoveBatchDim(ptr *Ctensor, self Ctensor, level int64, batchSize int64, outDim int64)
func Atg_ReshapeAlias(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, ...)
func Atg_ReshapeAliasCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, ...)
func Atg_ReshapeAliasCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, ...)
func Atg_ReshapeCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func Atg_ReshapeFromTensor(ptr *Ctensor, self Ctensor, shape Ctensor)
func Atg_ResizeOutput(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, device int32)
func Atg_ResizeOutputOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, ...)
func Atg_ResizeOutput_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, device int32)
func Atg_RowwisePrune(ptr *Ctensor, weight Ctensor, mask Ctensor, compressedIndicesDtype int32)
func Atg_SampleDirichlet(ptr *Ctensor, self Ctensor)
func Atg_SampleDirichletOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_SaturateWeightToFp16(ptr *Ctensor, weight Ctensor)
func Atg_ScaledDotProductAttentionMath(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, ...)
func Atg_ScaledDotProductEfficientAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnBias Ctensor, ...)
func Atg_ScaledDotProductFlashAttentionBackward(ptr *Ctensor, gradOut Ctensor, query Ctensor, key Ctensor, value Ctensor, ...)
func Atg_ScaledMm(ptr *Ctensor, self Ctensor, mat2 Ctensor, bias Ctensor, outDtype int32, ...)
func Atg_ScaledMmOut(ptr *Ctensor, out Ctensor, outAmax Ctensor, self Ctensor, mat2 Ctensor, ...)
func Atg_ScatterReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, ...)
func Atg_ScatterReduceTwoOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, ...)
func Atg_ScatterReduce_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, ...)
func Atg_SegmentReduceBackward(ptr *Ctensor, grad Ctensor, output Ctensor, data Ctensor, reduce string, ...)
func Atg_SegmentReduceBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, output Ctensor, data Ctensor, ...)
func Atg_ShapeAsTensor(ptr *Ctensor, self Ctensor)
func Atg_SlowConv2dBackward(ptr *Ctensor, gradInput Ctensor, gradWeight Ctensor, gradBias Ctensor, ...)
func Atg_SobolEngineDraw(ptr *Ctensor, quasi Ctensor, n int64, sobolstate Ctensor, dimension int64, ...)
func Atg_SobolEngineFf_(ptr *Ctensor, self Ctensor, n int64, sobolstate Ctensor, dimension int64, ...)
func Atg_SobolEngineInitializeState_(ptr *Ctensor, self Ctensor, dimension int64)
func Atg_SobolEngineScramble_(ptr *Ctensor, self Ctensor, ltm Ctensor, dimension int64)
func Atg_Softmax(ptr *Ctensor, self Ctensor, dim int64, halfToFloat int32)
func Atg_SoftmaxBackwardData(ptr *Ctensor, gradOutput Ctensor, output Ctensor, dim int64, inputDtype int32)
func Atg_SoftmaxBackwardDataOut(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, output Ctensor, dim int64, ...)
func Atg_SoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, halfToFloat int32)
func Atg_SparseAddmm(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)
func Atg_SparseAddmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)
func Atg_SparseBroadcastTo(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func Atg_SparseBroadcastToCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func Atg_SparseBroadcastToCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func Atg_SparseBscTensorUnsafe(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, ...)
func Atg_SparseBsrTensorUnsafe(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, ...)
func Atg_SparseCompressedTensorUnsafe(ptr *Ctensor, compressedIndices Ctensor, plainIndices Ctensor, values Ctensor, ...)
func Atg_SparseCooTensorUnsafe(ptr *Ctensor, indices Ctensor, values Ctensor, sizeData []int64, sizeLen int, ...)
func Atg_SparseCooTensorWithDims(ptr *Ctensor, sparseDim int64, denseDim int64, sizeData []int64, sizeLen int, ...)
func Atg_SparseCooTensorWithDimsAndTensors(ptr *Ctensor, sparseDim int64, denseDim int64, sizeData []int64, sizeLen int, ...)
func Atg_SparseCooTensorWithDimsAndTensorsOut(ptr *Ctensor, out Ctensor, sparseDim int64, denseDim int64, sizeData []int64, ...)
func Atg_SparseCooTensorWithDimsOut(ptr *Ctensor, out Ctensor, sparseDim int64, denseDim int64, sizeData []int64, ...)
func Atg_SparseCscTensorUnsafe(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, ...)
func Atg_SparseCsrProd(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, ...)
func Atg_SparseCsrProdDimDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func Atg_SparseCsrSum(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, ...)
func Atg_SparseCsrSumDimDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, ...)
func Atg_SparseCsrTensorUnsafe(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, ...)
func Atg_SparseLogSoftmax(ptr *Ctensor, self Ctensor, dim int64, halfToFloat int32)
func Atg_SparseLogSoftmaxBackwardData(ptr *Ctensor, gradOutput Ctensor, output Ctensor, dim int64, self Ctensor)
func Atg_SparseLogSoftmaxBackwardDataOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, dim int64, ...)
func Atg_SparseLogSoftmaxInt(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func Atg_SparseLogSoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, halfToFloat int32)
func Atg_SparseMaskProjection(ptr *Ctensor, self Ctensor, mask Ctensor, accumulateMatches int32)
func Atg_SparseMaskProjectionOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, accumulateMatches int32)
func Atg_SparseMm(ptr *Ctensor, sparse Ctensor, dense Ctensor)
func Atg_SparseMmReduce(ptr *Ctensor, sparse Ctensor, dense Ctensor, reduce string)
func Atg_SparseMmReduceImpl(ptr *Ctensor, self Ctensor, other Ctensor, reduce string)
func Atg_SparseSemiStructuredLinear(ptr *Ctensor, input Ctensor, weight Ctensor, meta Ctensor, bias Ctensor, ...)
func Atg_SparseSoftmax(ptr *Ctensor, self Ctensor, dim int64, halfToFloat int32)
func Atg_SparseSoftmaxBackwardData(ptr *Ctensor, gradOutput Ctensor, output Ctensor, dim int64, self Ctensor)
func Atg_SparseSoftmaxBackwardDataOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, dim int64, ...)
func Atg_SparseSoftmaxInt(ptr *Ctensor, self Ctensor, dim int64, dtype int32)
func Atg_SparseSoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, halfToFloat int32)
func Atg_SparseSparseMatmul(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg_SparseSparseMatmulOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func Atg_SparseSum(ptr *Ctensor, self Ctensor)
func Atg_SparseSumBackward(ptr *Ctensor, grad Ctensor, self Ctensor, dimData []int64, dimLen int)
func Atg_SparseSumBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, self Ctensor, dimData []int64, ...)
func Atg_SparseSumDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)
func Atg_SparseSumDimDtype(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, dtype int32)
func Atg_SparseSumDimOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int)
func Atg_SparseSumDtype(ptr *Ctensor, self Ctensor, dtype int32)
func Atg_Spdiags(ptr *Ctensor, diagonals Ctensor, offsets Ctensor, shapeData []int64, ...)
func Atg_SpdiagsOut(ptr *Ctensor, out Ctensor, diagonals Ctensor, offsets Ctensor, ...)
func Atg_Stack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func Atg_StackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)
func Atg_StandardGamma(ptr *Ctensor, self Ctensor)
func Atg_StandardGammaGrad(ptr *Ctensor, self Ctensor, output Ctensor)
func Atg_StandardGammaGradOut(ptr *Ctensor, out Ctensor, self Ctensor, output Ctensor)
func Atg_StandardGammaOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_TestAmbiguousDefaults(ptr *Ctensor, dummy Ctensor, a int64, b int64)
func Atg_TestAmbiguousDefaultsB(ptr *Ctensor, dummy Ctensor, a int64, b string)
func Atg_TestAutogradMultipleDispatch(ptr *Ctensor, self Ctensor)
func Atg_TestAutogradMultipleDispatchFullcoverageOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_TestAutogradMultipleDispatchNtonly(ptr *Ctensor, self Ctensor, b int32)
func Atg_TestAutogradMultipleDispatchView(ptr *Ctensor, self Ctensor)
func Atg_TestAutogradMultipleDispatchViewCopy(ptr *Ctensor, self Ctensor)
func Atg_TestAutogradMultipleDispatchViewCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_TestCheckTensor(ptr *Ctensor, self Ctensor)
func Atg_TestFunctorchFallback(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg_TestFunctorchFallbackOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func Atg_TestOptionalFilledIntlist(ptr *Ctensor, values Ctensor, addendsData []int64, addendsLen int)
func Atg_TestOptionalFilledIntlistOut(ptr *Ctensor, out Ctensor, values Ctensor, addendsData []int64, addendsLen int)
func Atg_TestOptionalFloatlist(ptr *Ctensor, values Ctensor, addendsData []float64, addendsLen int)
func Atg_TestOptionalFloatlistOut(ptr *Ctensor, out Ctensor, values Ctensor, addendsData []float64, ...)
func Atg_TestOptionalIntlist(ptr *Ctensor, values Ctensor, addendsData []int64, addendsLen int)
func Atg_TestOptionalIntlistOut(ptr *Ctensor, out Ctensor, values Ctensor, addendsData []int64, addendsLen int)
func Atg_TestSerializationSubcmul(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg_TestStringDefault(ptr *Ctensor, dummy Ctensor, a string, b string)
func Atg_TestWarnInAutograd(ptr *Ctensor, self Ctensor)
func Atg_TestWarnInAutogradOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_ToCopy(ptr *Ctensor, self Ctensor, optionsKind int32, optionsDevice int32, ...)
func Atg_ToCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, nonBlocking int32)
func Atg_ToDense(ptr *Ctensor, self Ctensor, dtype int32, maskedGrad int32)
func Atg_ToDenseOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32, maskedGrad int32)
func Atg_ToSparse(ptr *Ctensor, self Ctensor, layout int8, blocksizeData []int64, ...)
func Atg_ToSparseBsc(ptr *Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, ...)
func Atg_ToSparseBscOut(ptr *Ctensor, out Ctensor, self Ctensor, blocksizeData []int64, ...)
func Atg_ToSparseBsr(ptr *Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, ...)
func Atg_ToSparseBsrOut(ptr *Ctensor, out Ctensor, self Ctensor, blocksizeData []int64, ...)
func Atg_ToSparseCsc(ptr *Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)
func Atg_ToSparseCscOut(ptr *Ctensor, out Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)
func Atg_ToSparseCsr(ptr *Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)
func Atg_ToSparseCsrOut(ptr *Ctensor, out Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)
func Atg_ToSparseOut(ptr *Ctensor, out Ctensor, self Ctensor, layout int8, blocksizeData []int64, ...)
func Atg_ToSparseSemiStructured(ptr *Ctensor, dense Ctensor)
func Atg_ToSparseSparseDim(ptr *Ctensor, self Ctensor, sparseDim int64)
func Atg_ToSparseSparseDimOut(ptr *Ctensor, out Ctensor, self Ctensor, sparseDim int64)
func Atg_TransformBiasRescaleQkv(ptr *Ctensor, qkv Ctensor, qkvBias Ctensor, numHeads int64)
func Atg_TransformBiasRescaleQkvOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, qkv Ctensor, ...)
func Atg_TransformerEncoderLayerFwd(ptr *Ctensor, src Ctensor, embedDim int64, numHeads int64, qkvWeight Ctensor, ...)
func Atg_TransformerEncoderLayerFwdOut(ptr *Ctensor, out Ctensor, src Ctensor, embedDim int64, numHeads int64, ...)
func Atg_Trilinear(ptr *Ctensor, i1 Ctensor, i2 Ctensor, i3 Ctensor, expand1Data []int64, ...)
func Atg_TrilinearOut(ptr *Ctensor, out Ctensor, i1 Ctensor, i2 Ctensor, i3 Ctensor, ...)
func Atg_TritonMultiHeadAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, embedDim int64, ...)
func Atg_TritonMultiHeadAttentionOut(ptr *Ctensor, out Ctensor, query Ctensor, key Ctensor, value Ctensor, ...)
func Atg_TritonScaledDotAttention(ptr *Ctensor, q Ctensor, k Ctensor, v Ctensor, dropoutP float64)
func Atg_TritonScaledDotAttentionOut(ptr *Ctensor, out Ctensor, q Ctensor, k Ctensor, v Ctensor, dropoutP float64)
func Atg_Unique(ptr *Ctensor, self Ctensor, sorted int32, returnInverse int32)
func Atg_Unique2(ptr *Ctensor, self Ctensor, sorted int32, returnInverse int32, ...)
func Atg_Unique2Out(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, self Ctensor, ...)
func Atg_UniqueOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, sorted int32, ...)
func Atg_UnpackDual(ptr *Ctensor, dual Ctensor, level int64)
func Atg_UnsafeIndex(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int)
func Atg_UnsafeIndexPut(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, ...)
func Atg_UnsafeView(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func Atg_UnsafeViewOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)
func Atg_UpsampleBicubic2dAa(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleBicubic2dAaBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleBicubic2dAaBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleBicubic2dAaOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleBicubic2dAaVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleBilinear2dAa(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleBilinear2dAaBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleBilinear2dAaBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleBilinear2dAaOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleBilinear2dAaVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleNearestExact1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleNearestExact1dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleNearestExact1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleNearestExact1dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleNearestExact1dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleNearestExact2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleNearestExact2dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleNearestExact2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleNearestExact2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleNearestExact2dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleNearestExact3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleNearestExact3dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UpsampleNearestExact3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleNearestExact3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, ...)
func Atg_UpsampleNearestExact3dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, ...)
func Atg_UseCudnnCtcLoss(logProbs Ctensor, targets Ctensor, inputLengthsData []int64, ...) bool
func Atg_UseCudnnCtcLossTensor(logProbs Ctensor, targets Ctensor, inputLengths Ctensor, targetLengths Ctensor, ...) bool
func Atg_UseCudnnRnnFlattenWeight() bool
func Atg_Values(ptr *Ctensor, self Ctensor)
func Atg_ValuesCopy(ptr *Ctensor, self Ctensor)
func Atg_ValuesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)
func Atg_Version(self Ctensor) int64
func Atg_WeightNorm(ptr *Ctensor, v Ctensor, g Ctensor, dim int64)
func Atg_WeightNormDifferentiableBackward(ptr *Ctensor, gradW Ctensor, savedV Ctensor, savedG Ctensor, ...)
func Atg_WeightNormInterface(ptr *Ctensor, v Ctensor, g Ctensor, dim int64)
func Atg_WeightNormInterfaceBackward(ptr *Ctensor, gradW Ctensor, savedV Ctensor, savedG Ctensor, ...)
func Atg_WeightNormInterfaceBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, gradW Ctensor, savedV Ctensor, ...)
func Atg_WeightNormInterfaceOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, v Ctensor, g Ctensor, dim int64)
func Atg__AndTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__And_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg__IandTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__Iand_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg__IlshiftTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__Ilshift_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg__IorTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__Ior_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg__IrshiftTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__Irshift_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg__IxorTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__Ixor_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg__LshiftScalarOut_(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func Atg__LshiftTensorOut_(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func Atg__LshiftTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__Lshift_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg__OrTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__Or_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg__RshiftScalarOut_(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)
func Atg__RshiftTensorOut_(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)
func Atg__RshiftTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__Rshift_(ptr *Ctensor, self Ctensor, other Cscalar)
func Atg__XorTensor_(ptr *Ctensor, self Ctensor, other Ctensor)
func Atg__Xor_(ptr *Ctensor, self Ctensor, other Cscalar)
func AtiFree(val Civalue)
func AtiLength(val Civalue) int32
func AtiTag(val Civalue) int32
func AtiToBool(val Civalue) bool
func AtiToBoolList(val Civalue, ptr unsafe.Pointer, boolListLen int)
func AtiToDouble(val Civalue) float64
func AtiToDoubleList(val Civalue, ptr unsafe.Pointer, doubleListLen int)
func AtiToGenericDict(val Civalue, ptr *Civalue, genericDictLen int)
func AtiToGenericList(val Civalue, ptr *Civalue, genericListLen int)
func AtiToInt(val Civalue) int64
func AtiToIntList(val Civalue, ptr unsafe.Pointer, intListLen int)
func AtiToString(val Civalue) string
func AtiToTensorList(val Civalue, ptr *Ctensor, tensorListLen int)
func AtiToTuple(val Civalue, ptr *Civalue, tupleLen int)
func AtiTupleLength(val Civalue) int32
func AtmEval(m Cmodule)
func AtmFree(m Cmodule)
func AtmGetProfilingMode() bool
func AtmNamedParameters(m Cmodule, dataPtr unsafe.Pointer)
func AtmSave(m Cmodule, path string)
func AtmSetProfilingMode(b bool)
func AtmTo(m Cmodule, device int32, dtype int32, nonBlocking bool)
func AtmTrain(m Cmodule)
func AtoAddParamGroup(coptimizer Coptimizer, tensors []Ctensor, ntensors int)
func AtoAddParameter(coptimizer Coptimizer, tensor Ctensor, group uint)
func AtoAddParametersOld(coptimizer Coptimizer, tensors []Ctensor, ntensors int)
func AtoConstantPadNd(ptr *Ctensor, self Ctensor, padData []int64, padLen int, value Cscalar)
func AtoFree(coptimizer Coptimizer)
func AtoGetLearningRates(coptimizer Coptimizer) []float64
func AtoParamGroupNum(coptimizer Coptimizer) int64
func AtoSetLearningRate(coptimizer Coptimizer, learningRate float64)
func AtoSetLearningRates(coptimizer Coptimizer, lrs []float64)
func AtoSetMomentum(coptimizer Coptimizer, momentum float64)
func AtoStep(coptimizer Coptimizer)
func AtoZeroGrad(coptimizer Coptimizer)
func AtsFree(cscalar Cscalar)
func AtsToFloat(cscalar Cscalar) float64
func AtsToInt(cscalar Cscalar) int64
func AtsToString(cscalar Cscalar) string
func GetAndResetLastErr() *C.char
func IsNull(ctensor Ctensor) bool
type Civalue
- func AtiBool(val bool) Civalue
- func AtiBoolList(boolListData []bool, boolListLen int) Civalue
- func AtiDouble(val float64) Civalue
- func AtiDoubleList(doubleListData []float64, doubleListLen int) Civalue
- func AtiGenericDict(genericDictData []Civalue, genericDictLen int) Civalue
- func AtiGenericList(genericListData []Civalue, genericListLen int) Civalue
- func AtiInt(val int64) Civalue
- func AtiIntList(intListData []int64, intListLen int) Civalue
- func AtiNone() Civalue
- func AtiString(val string) Civalue
- func AtiTensor(ts Ctensor) Civalue
- func AtiTensorList(tensorListData []Ctensor, tensorListLen int) Civalue
- func AtiTuple(tupleData []Civalue, tupleLen int) Civalue
- func AtmForward_(m Cmodule, ivalues *Civalue, nivalues int) Civalue
type Cmodule
- func AtmLoad(path string) Cmodule
- func AtmLoadOnDevice(path string, device int32) Cmodule
- func AtmLoadStr(val string, sz int) Cmodule
- func AtmLoadStrOnDevice(val string, sz int, device int32) Cmodule
type Coptimizer
- func AtoAdam(learningRate, beta1, beta2, weightDecay float64) Coptimizer
- func AtoAdamW(learningRate, beta1, beta2, weightDecay float64) Coptimizer
- func AtoRmsProp(learningRate, alpha, eps, weightDecay, momentum float64, centered int) Coptimizer
- func AtoSgd(learningRate, momentum, dampening, weightDecay float64, nesterov int) Coptimizer
type Cscalar
- func AtsFloat(v float64) Cscalar
- func AtsInt(v int64) Cscalar
type Ctensor
- func AtGet(ts Ctensor, index int) Ctensor
- func AtLoad(path string) Ctensor
- func AtLoadImage(path string) Ctensor
- func AtNewTensor() Ctensor
- func AtResizeImage(ts Ctensor, w, h int64) Ctensor
- func AtShallowClone(ts Ctensor) Ctensor
- func AtTensorOfData(vs unsafe.Pointer, dims []int64, ndims uint, elt_size_in_bytes uint, kind int) Ctensor
- func AtgAlignTensors(tensorsData []Ctensor, tensorsLen int) *Ctensor
- func AtgBroadcastTensors(tensorsData []Ctensor, tensorsLen int) *Ctensor
- func AtgChunk(self Ctensor, chunks int64, dim int64) *Ctensor
- func AtgMeshgrid(tensorsData []Ctensor, tensorsLen int) *Ctensor
- func AtgNonzeroNumpy(self Ctensor) *Ctensor
- func AtgSplit(self Ctensor, splitSize int64, dim int64) *Ctensor
- func AtgSplitWithSizes(self Ctensor, splitSizesData []int64, splitSizesLen int, dim int64) *Ctensor
- func AtgUnbind(self Ctensor, dim int64) *Ctensor
- func AtgWhere(condition Ctensor) *Ctensor
- func AtiToTensor(val Civalue) Ctensor
- func AtmForward(m Cmodule, tensors *Ctensor, ntensors int) Ctensor
- func NewTensor() Ctensor
type LoadData
type NamedCtensor
type PointerStore
- func NewPointerStore() PointerStore
- func (ps PointerStore) Free(ptr unsafe.Pointer)
- func (ps PointerStore) Get(ptr unsafe.Pointer) (v interface{})
- func (ps PointerStore) Set(v interface{}) unsafe.Pointer

Constants ¶

This section is empty.

Variables ¶

View Source

var PStore = NewPointerStore()

Functions ¶

func AtBackward ¶

func AtBackward(ts Ctensor, keepGraph int, createGraph int)

void at_backward(tensor, int, int);

func AtCopyData ¶

func AtCopyData(ts Ctensor, vs unsafe.Pointer, numel uint, element_size_in_bytes uint)

void at_copy_data(tensor tensor, void *vs, size_t numel, size_t element_size_in_bytes);

func AtCopy_ ¶

func AtCopy_(dst Ctensor, src Ctensor)

void at_copy_(tensor dst, tensor src);

func AtDataPtr ¶

func AtDataPtr(t Ctensor) unsafe.Pointer

void *at_data_ptr(tensor);

func AtDefined ¶

func AtDefined(ts Ctensor) bool

int at_defined(tensor);

func AtDevice ¶

func AtDevice(ts Ctensor) int

int at_device(tensor);

func AtDim ¶

func AtDim(t Ctensor) uint64

size_t at_dim(tensor);

func AtDoubleValueAtIndexes ¶

func AtDoubleValueAtIndexes(ts Ctensor, indexes unsafe.Pointer, indexesLen int) float64

double at_double_value_at_indexes(tensor, int64_t *indexes, int indexes_len);

func AtFree ¶

func AtFree(ts Ctensor)

void at_free(tensor);

func AtGradSetEnabled ¶

func AtGradSetEnabled(b int) int

int at_grad_set_enabled(int b);

func AtInt64ValueAtIndexes ¶

func AtInt64ValueAtIndexes(ts Ctensor, indexes unsafe.Pointer, indexesLen int) int64

int64_t at_int64_value_at_indexes(tensor, int64_t *indexes, int indexes_len);

func AtIsContiguous ¶ added in v0.8.0

func AtIsContiguous(ts Ctensor) bool

int at_is_contiguous(tensor);

func AtIsMkldnn ¶ added in v0.8.0

func AtIsMkldnn(ts Ctensor) bool

int at_is_mkldnn(tensor);

func AtIsSparse ¶

func AtIsSparse(ts Ctensor) bool

int at_is_sparse(tensor);

func AtLoadCallback ¶

func AtLoadCallback(filename string, dataPtr unsafe.Pointer)

void at_load_callback(char *filename, void *data, void (*f)(void *, char *, tensor));

void at_load_callback(char *filename, void *data, void (*f)(void *, char *, tensor)) {
PROTECT(
auto module = torch::jit::load(filename);
for (const auto &p : module.named_parameters()) {
auto v = p.value;
f(data, (char*)p.name.c_str(), new torch::Tensor(v));
}
)
} *

func AtLoadCallbackWithDevice ¶

func AtLoadCallbackWithDevice(filename string, dataPtr unsafe.Pointer, device int32)

* void at_load_callback_with_device(char *filename, void *data, void (*f)(void *, char *, tensor), int device_id) { * PROTECT( * auto module = torch::jit::load(filename, device_of_int(device_id)); * for (const auto &p : module.named_parameters()) { * auto v = p.value; * f(data, (char*)p.name.c_str(), new torch::Tensor(v)); * } * ) * } *

func AtLoadMulti ¶

func AtLoadMulti(tensors []Ctensor, tensor_names []string, ntensors int, filename string)

[at_load_multi] takes as input an array of nullptr for [tensors].

void at_load_multi(tensor *tensors, char **tensor_names, int ntensors, char *filename);

func AtPrint ¶

func AtPrint(t Ctensor)

void at_print(tensor);

func AtRequiresGrad ¶

func AtRequiresGrad(ts Ctensor) bool

int at_requires_grad(tensor);

func AtRunBackward ¶

func AtRunBackward(tensorsPtr *Ctensor, ntensors int, inputsPtr *Ctensor, ninputs int, outputsPtr *Ctensor, keepGraph int, createGraph int)

* void at_run_backward(tensor *tensors, * int ntensors, * tensor *inputs, * int ninputs, * tensor *outputs, * int keep_graph, * int create_graph); *

func AtSave ¶

func AtSave(ts Ctensor, path string)

void at_save(tensor, char *filename);

func AtSaveImage ¶

func AtSaveImage(ts Ctensor, path string)

int at_save_image(tensor, char *filename);

func AtSaveMulti ¶

func AtSaveMulti(tensors []Ctensor, tensor_names []string, ntensors int, filename string)

void at_save_multi(tensor *tensors, char **tensor_names, int ntensors, char *filename);

func AtSaveMultiNew ¶ added in v0.1.10

func AtSaveMultiNew(tensors []Ctensor, names []string, filename string)

void at_save_multi(tensor *tensors, char **tensor_names, int ntensors, char *filename);

func AtScalarType ¶

func AtScalarType(t Ctensor) int32

int at_scalar_type(tensor);

func AtShape ¶

func AtShape(t Ctensor, ptr unsafe.Pointer)

void at_shape(tensor, int64_t *);

func AtStride ¶ added in v0.8.0

func AtStride(t Ctensor, ptr unsafe.Pointer)

void at_stride(tensor, int64_t *);

func AtToString ¶

func AtToString(ts Ctensor, lineSize int64) string

* char *at_to_string(tensor t, int line_size) { * PROTECT( * std::ostringstream oss; * torch::print(oss, *t, line_size); * return strdup(oss.str().c_str()); * ) * return nullptr; * } *

func AtcCudaDeviceCount ¶

func AtcCudaDeviceCount() int32

int atc_cuda_device_count();

func AtcCudaIsAvailable ¶

func AtcCudaIsAvailable() bool

int atc_cuda_is_available();

func AtcCudnnIsAvailable ¶

func AtcCudnnIsAvailable() bool

int atc_cudnn_is_available();

func AtcGetDevice ¶ added in v0.8.0

func AtcGetDevice() int

int atc_get_device();

func AtcSetBenchmarkCudnn ¶

func AtcSetBenchmarkCudnn(b int)

void atc_set_benchmark_cudnn(int b);

func AtcSetDevice ¶ added in v0.8.0

func AtcSetDevice(deviceIndex int) int

int atc_set_device(int device_index);

func AtcSynchronize ¶ added in v0.8.0

func AtcSynchronize(deviceIndex int64)

void atc_synchronize(int64_t device_index);

func Atc_cuda_device_count ¶

func Atc_cuda_device_count() int

func AtgAbs ¶

func AtgAbs(ptr *Ctensor, self Ctensor)

func AtgAbsOut ¶

func AtgAbsOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAbs_ ¶

func AtgAbs_(ptr *Ctensor, self Ctensor)

func AtgAbsolute ¶ added in v0.3.0

func AtgAbsolute(ptr *Ctensor, self Ctensor)

func AtgAbsoluteOut ¶ added in v0.3.0

func AtgAbsoluteOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAbsolute_ ¶ added in v0.3.0

func AtgAbsolute_(ptr *Ctensor, self Ctensor)

func AtgAcos ¶

func AtgAcos(ptr *Ctensor, self Ctensor)

func AtgAcosOut ¶

func AtgAcosOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAcos_ ¶

func AtgAcos_(ptr *Ctensor, self Ctensor)

func AtgAcosh ¶ added in v0.3.0

func AtgAcosh(ptr *Ctensor, self Ctensor)

func AtgAcoshOut ¶ added in v0.3.0

func AtgAcoshOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAcosh_ ¶ added in v0.3.0

func AtgAcosh_(ptr *Ctensor, self Ctensor)

func AtgAdaptiveAvgPool1d ¶

func AtgAdaptiveAvgPool1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdaptiveAvgPool2d ¶

func AtgAdaptiveAvgPool2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdaptiveAvgPool2dOut ¶

func AtgAdaptiveAvgPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdaptiveAvgPool3d ¶

func AtgAdaptiveAvgPool3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdaptiveAvgPool3dBackward ¶

func AtgAdaptiveAvgPool3dBackward(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor)

func AtgAdaptiveAvgPool3dOut ¶

func AtgAdaptiveAvgPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdaptiveMaxPool1d ¶

func AtgAdaptiveMaxPool1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdaptiveMaxPool2d ¶

func AtgAdaptiveMaxPool2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdaptiveMaxPool2dBackward ¶

func AtgAdaptiveMaxPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, indices Ctensor)

func AtgAdaptiveMaxPool2dBackwardGradInput ¶ added in v0.4.0

func AtgAdaptiveMaxPool2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, indices Ctensor)

func AtgAdaptiveMaxPool2dOut ¶

func AtgAdaptiveMaxPool2dOut(ptr *Ctensor, out Ctensor, indices Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdaptiveMaxPool3d ¶

func AtgAdaptiveMaxPool3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdaptiveMaxPool3dBackward ¶

func AtgAdaptiveMaxPool3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, indices Ctensor)

func AtgAdaptiveMaxPool3dBackwardGradInput ¶ added in v0.4.0

func AtgAdaptiveMaxPool3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, indices Ctensor)

func AtgAdaptiveMaxPool3dOut ¶

func AtgAdaptiveMaxPool3dOut(ptr *Ctensor, out Ctensor, indices Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgAdd ¶

func AtgAdd(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgAddOut ¶

func AtgAddOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgAddScalar ¶ added in v0.4.0

func AtgAddScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgAddScalarOut ¶ added in v0.8.0

func AtgAddScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgAddScalar_ ¶ added in v0.4.0

func AtgAddScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgAdd_ ¶

func AtgAdd_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgAddbmm ¶

func AtgAddbmm(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)

func AtgAddbmmOut ¶

func AtgAddbmmOut(ptr *Ctensor, out Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)

func AtgAddbmm_ ¶

func AtgAddbmm_(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)

func AtgAddcdiv ¶

func AtgAddcdiv(ptr *Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)

func AtgAddcdivOut ¶

func AtgAddcdivOut(ptr *Ctensor, out Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)

func AtgAddcdiv_ ¶

func AtgAddcdiv_(ptr *Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)

func AtgAddcmul ¶

func AtgAddcmul(ptr *Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)

func AtgAddcmulOut ¶

func AtgAddcmulOut(ptr *Ctensor, out Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)

func AtgAddcmul_ ¶

func AtgAddcmul_(ptr *Ctensor, self Ctensor, tensor1 Ctensor, tensor2 Ctensor)

func AtgAddmm ¶

func AtgAddmm(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)

func AtgAddmmOut ¶

func AtgAddmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)

func AtgAddmm_ ¶

func AtgAddmm_(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)

func AtgAddmv ¶

func AtgAddmv(ptr *Ctensor, self Ctensor, mat Ctensor, vec Ctensor)

func AtgAddmvOut ¶

func AtgAddmvOut(ptr *Ctensor, out Ctensor, self Ctensor, mat Ctensor, vec Ctensor)

func AtgAddmv_ ¶

func AtgAddmv_(ptr *Ctensor, self Ctensor, mat Ctensor, vec Ctensor)

func AtgAddr ¶

func AtgAddr(ptr *Ctensor, self Ctensor, vec1 Ctensor, vec2 Ctensor)

func AtgAddrOut ¶

func AtgAddrOut(ptr *Ctensor, out Ctensor, self Ctensor, vec1 Ctensor, vec2 Ctensor)

func AtgAddr_ ¶

func AtgAddr_(ptr *Ctensor, self Ctensor, vec1 Ctensor, vec2 Ctensor)

func AtgAdjoint ¶ added in v0.7.0

func AtgAdjoint(ptr *Ctensor, self Ctensor)

func AtgAffineGridGenerator ¶

func AtgAffineGridGenerator(ptr *Ctensor, theta Ctensor, sizeData []int64, sizeLen int, alignCorners int32)

func AtgAffineGridGeneratorBackward ¶

func AtgAffineGridGeneratorBackward(ptr *Ctensor, grad Ctensor, sizeData []int64, sizeLen int, alignCorners int32)

func AtgAffineGridGeneratorOut ¶ added in v0.8.0

func AtgAffineGridGeneratorOut(ptr *Ctensor, out Ctensor, theta Ctensor, sizeData []int64, sizeLen int, alignCorners int32)

func AtgAlias ¶

func AtgAlias(ptr *Ctensor, self Ctensor)

func AtgAliasCopy ¶ added in v0.8.0

func AtgAliasCopy(ptr *Ctensor, self Ctensor)

func AtgAliasCopyOut ¶ added in v0.8.0

func AtgAliasCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAlignAs ¶

func AtgAlignAs(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgAll ¶

func AtgAll(ptr *Ctensor, self Ctensor)

func AtgAllAllOut ¶ added in v0.5.0

func AtgAllAllOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAllDim ¶ added in v0.4.0

func AtgAllDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgAllOut ¶

func AtgAllOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgAllclose ¶ added in v0.4.0

func AtgAllclose(self Ctensor, other Ctensor, rtol float64, atol float64, equalNan int32) bool

func AtgAlphaDropout ¶

func AtgAlphaDropout(ptr *Ctensor, input Ctensor, p float64, train int32)

func AtgAlphaDropout_ ¶

func AtgAlphaDropout_(ptr *Ctensor, self Ctensor, p float64, train int32)

func AtgAmax ¶ added in v0.3.0

func AtgAmax(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgAmaxOut ¶ added in v0.3.0

func AtgAmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgAmin ¶ added in v0.3.0

func AtgAmin(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgAminOut ¶ added in v0.3.0

func AtgAminOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgAminmax ¶ added in v0.5.0

func AtgAminmax(ptr *Ctensor, self Ctensor, dimVal int64, dimNull int, keepdim int32)

func AtgAminmaxOut ¶ added in v0.5.0

func AtgAminmaxOut(ptr *Ctensor, min Ctensor, max Ctensor, self Ctensor, dimVal int64, dimNull int, keepdim int32)

func AtgAngle ¶

func AtgAngle(ptr *Ctensor, self Ctensor)

func AtgAngleOut ¶

func AtgAngleOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAny ¶

func AtgAny(ptr *Ctensor, self Ctensor)

func AtgAnyAllOut ¶ added in v0.5.0

func AtgAnyAllOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAnyDim ¶ added in v0.4.0

func AtgAnyDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgAnyOut ¶

func AtgAnyOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgArange ¶

func AtgArange(ptr *Ctensor, end Cscalar, optionsKind int32, optionsDevice int32)

func AtgArangeStart ¶ added in v0.4.0

func AtgArangeStart(ptr *Ctensor, start Cscalar, end Cscalar, optionsKind int32, optionsDevice int32)

func AtgArangeStartStep ¶ added in v0.4.0

func AtgArangeStartStep(ptr *Ctensor, start Cscalar, end Cscalar, step Cscalar, optionsKind int32, optionsDevice int32)

func AtgArccos ¶ added in v0.3.0

func AtgArccos(ptr *Ctensor, self Ctensor)

func AtgArccosOut ¶ added in v0.3.0

func AtgArccosOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgArccos_ ¶ added in v0.3.0

func AtgArccos_(ptr *Ctensor, self Ctensor)

func AtgArccosh ¶ added in v0.3.0

func AtgArccosh(ptr *Ctensor, self Ctensor)

func AtgArccoshOut ¶ added in v0.3.0

func AtgArccoshOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgArccosh_ ¶ added in v0.3.0

func AtgArccosh_(ptr *Ctensor, self Ctensor)

func AtgArcsin ¶ added in v0.3.0

func AtgArcsin(ptr *Ctensor, self Ctensor)

func AtgArcsinOut ¶ added in v0.3.0

func AtgArcsinOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgArcsin_ ¶ added in v0.3.0

func AtgArcsin_(ptr *Ctensor, self Ctensor)

func AtgArcsinh ¶ added in v0.3.0

func AtgArcsinh(ptr *Ctensor, self Ctensor)

func AtgArcsinhOut ¶ added in v0.3.0

func AtgArcsinhOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgArcsinh_ ¶ added in v0.3.0

func AtgArcsinh_(ptr *Ctensor, self Ctensor)

func AtgArctan ¶ added in v0.3.0

func AtgArctan(ptr *Ctensor, self Ctensor)

func AtgArctan2 ¶ added in v0.7.0

func AtgArctan2(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgArctan2Out ¶ added in v0.7.0

func AtgArctan2Out(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgArctan2_ ¶ added in v0.7.0

func AtgArctan2_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgArctanOut ¶ added in v0.3.0

func AtgArctanOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgArctan_ ¶ added in v0.3.0

func AtgArctan_(ptr *Ctensor, self Ctensor)

func AtgArctanh ¶ added in v0.3.0

func AtgArctanh(ptr *Ctensor, self Ctensor)

func AtgArctanhOut ¶ added in v0.3.0

func AtgArctanhOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgArctanh_ ¶ added in v0.3.0

func AtgArctanh_(ptr *Ctensor, self Ctensor)

func AtgArgmax ¶

func AtgArgmax(ptr *Ctensor, self Ctensor, dimVal int64, dimNull int, keepdim int32)

func AtgArgmaxOut ¶ added in v0.4.0

func AtgArgmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dimVal int64, dimNull int, keepdim int32)

func AtgArgmin ¶

func AtgArgmin(ptr *Ctensor, self Ctensor, dimVal int64, dimNull int, keepdim int32)

func AtgArgminOut ¶ added in v0.4.0

func AtgArgminOut(ptr *Ctensor, out Ctensor, self Ctensor, dimVal int64, dimNull int, keepdim int32)

func AtgArgsort ¶

func AtgArgsort(ptr *Ctensor, self Ctensor, dim int64, descending int32)

func AtgArgsortStable ¶ added in v0.8.0

func AtgArgsortStable(ptr *Ctensor, self Ctensor, stable int32, dim int64, descending int32)

func AtgArgsortStableOut ¶ added in v0.8.0

func AtgArgsortStableOut(ptr *Ctensor, out Ctensor, self Ctensor, stable int32, dim int64, descending int32)

func AtgArgwhere ¶ added in v0.7.0

func AtgArgwhere(ptr *Ctensor, self Ctensor)

func AtgAsStrided ¶

func AtgAsStrided(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int, storageOffsetVal int64, storageOffsetNull int)

func AtgAsStridedCopy ¶ added in v0.8.0

func AtgAsStridedCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int, storageOffsetVal int64, storageOffsetNull int)

func AtgAsStridedCopyOut ¶ added in v0.8.0

func AtgAsStridedCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int, storageOffsetVal int64, storageOffsetNull int)

func AtgAsStridedScatter ¶ added in v0.8.0

func AtgAsStridedScatter(ptr *Ctensor, self Ctensor, src Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int, storageOffsetVal int64, storageOffsetNull int)

func AtgAsStridedScatterOut ¶ added in v0.8.0

func AtgAsStridedScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int, storageOffsetVal int64, storageOffsetNull int)

func AtgAsStrided_ ¶

func AtgAsStrided_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int, storageOffsetVal int64, storageOffsetNull int)

func AtgAsin ¶

func AtgAsin(ptr *Ctensor, self Ctensor)

func AtgAsinOut ¶

func AtgAsinOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAsin_ ¶

func AtgAsin_(ptr *Ctensor, self Ctensor)

func AtgAsinh ¶ added in v0.3.0

func AtgAsinh(ptr *Ctensor, self Ctensor)

func AtgAsinhOut ¶ added in v0.3.0

func AtgAsinhOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAsinh_ ¶ added in v0.3.0

func AtgAsinh_(ptr *Ctensor, self Ctensor)

func AtgAtan ¶

func AtgAtan(ptr *Ctensor, self Ctensor)

func AtgAtan2 ¶

func AtgAtan2(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgAtan2Out ¶

func AtgAtan2Out(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgAtan2_ ¶

func AtgAtan2_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgAtanOut ¶

func AtgAtanOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAtan_ ¶

func AtgAtan_(ptr *Ctensor, self Ctensor)

func AtgAtanh ¶ added in v0.3.0

func AtgAtanh(ptr *Ctensor, self Ctensor)

func AtgAtanhOut ¶ added in v0.3.0

func AtgAtanhOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgAtanh_ ¶ added in v0.3.0

func AtgAtanh_(ptr *Ctensor, self Ctensor)

func AtgAtleast1d ¶ added in v0.3.0

func AtgAtleast1d(ptr *Ctensor, self Ctensor)

func AtgAtleast2d ¶ added in v0.3.0

func AtgAtleast2d(ptr *Ctensor, self Ctensor)

func AtgAtleast3d ¶ added in v0.3.0

func AtgAtleast3d(ptr *Ctensor, self Ctensor)

func AtgAvgPool1d ¶

func AtgAvgPool1d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32)

func AtgAvgPool2d ¶

func AtgAvgPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32, divisorOverrideVal int64, divisorOverrideNull int)

func AtgAvgPool2dBackward ¶

func AtgAvgPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32, divisorOverrideVal int64, divisorOverrideNull int)

func AtgAvgPool2dBackwardGradInput ¶ added in v0.4.0

func AtgAvgPool2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32, divisorOverrideVal int64, divisorOverrideNull int)

func AtgAvgPool2dOut ¶

func AtgAvgPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32, divisorOverrideVal int64, divisorOverrideNull int)

func AtgAvgPool3d ¶

func AtgAvgPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32, divisorOverrideVal int64, divisorOverrideNull int)

func AtgAvgPool3dBackward ¶

func AtgAvgPool3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32, divisorOverrideVal int64, divisorOverrideNull int)

func AtgAvgPool3dBackwardGradInput ¶ added in v0.4.0

func AtgAvgPool3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32, divisorOverrideVal int64, divisorOverrideNull int)

func AtgAvgPool3dOut ¶

func AtgAvgPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, ceilMode int32, countIncludePad int32, divisorOverrideVal int64, divisorOverrideNull int)

func AtgBaddbmm ¶

func AtgBaddbmm(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor, beta Cscalar, alpha Cscalar)

func AtgBaddbmmOut ¶

func AtgBaddbmmOut(ptr *Ctensor, out Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)

func AtgBaddbmm_ ¶

func AtgBaddbmm_(ptr *Ctensor, self Ctensor, batch1 Ctensor, batch2 Ctensor)

func AtgBartlettWindow ¶

func AtgBartlettWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)

func AtgBartlettWindowOut ¶ added in v0.8.0

func AtgBartlettWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)

func AtgBartlettWindowPeriodic ¶ added in v0.4.0

func AtgBartlettWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, optionsDevice int32)

func AtgBartlettWindowPeriodicOut ¶ added in v0.8.0

func AtgBartlettWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)

func AtgBatchNorm ¶

func AtgBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, momentum float64, eps float64, cudnnEnabled int32)

func AtgBatchNormBackwardElemt ¶

func AtgBatchNormBackwardElemt(ptr *Ctensor, gradOut Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, weight Ctensor, sumDy Ctensor, sumDyXmu Ctensor, count Ctensor)

func AtgBatchNormBackwardElemtOut ¶ added in v0.8.0

func AtgBatchNormBackwardElemtOut(ptr *Ctensor, out Ctensor, gradOut Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, weight Ctensor, sumDy Ctensor, sumDyXmu Ctensor, count Ctensor)

func AtgBatchNormBackwardReduce ¶

func AtgBatchNormBackwardReduce(ptr *Ctensor, gradOut Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, weight Ctensor, inputG int32, weightG int32, biasG int32)

func AtgBatchNormBackwardReduceOut ¶ added in v0.8.0

func AtgBatchNormBackwardReduceOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, gradOut Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, weight Ctensor, inputG int32, weightG int32, biasG int32)

func AtgBatchNormElemt ¶

func AtgBatchNormElemt(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, mean Ctensor, invstd Ctensor, eps float64)

func AtgBatchNormElemtOut ¶

func AtgBatchNormElemtOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, mean Ctensor, invstd Ctensor, eps float64)

func AtgBatchNormGatherStats ¶

func AtgBatchNormGatherStats(ptr *Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64, count int64)

func AtgBatchNormGatherStatsOut ¶ added in v0.8.0

func AtgBatchNormGatherStatsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64, count int64)

func AtgBatchNormGatherStatsWithCounts ¶

func AtgBatchNormGatherStatsWithCounts(ptr *Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64, counts Ctensor)

func AtgBatchNormGatherStatsWithCountsOut ¶ added in v0.8.0

func AtgBatchNormGatherStatsWithCountsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, mean Ctensor, invstd Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64, counts Ctensor)

func AtgBatchNormStats ¶

func AtgBatchNormStats(ptr *Ctensor, input Ctensor, eps float64)

func AtgBatchNormStatsOut ¶ added in v0.8.0

func AtgBatchNormStatsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, eps float64)

func AtgBatchNormUpdateStats ¶

func AtgBatchNormUpdateStats(ptr *Ctensor, input Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64)

func AtgBatchNormUpdateStatsOut ¶ added in v0.8.0

func AtgBatchNormUpdateStatsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64)

func AtgBernoulli ¶

func AtgBernoulli(ptr *Ctensor, self Ctensor)

func AtgBernoulliFloat_ ¶ added in v0.4.0

func AtgBernoulliFloat_(ptr *Ctensor, self Ctensor, p float64)

func AtgBernoulliP ¶ added in v0.4.0

func AtgBernoulliP(ptr *Ctensor, self Ctensor, p float64)

func AtgBernoulliTensor ¶ added in v0.8.0

func AtgBernoulliTensor(ptr *Ctensor, self Ctensor, p Ctensor)

func AtgBernoulli_ ¶

func AtgBernoulli_(ptr *Ctensor, self Ctensor, p Ctensor)

func AtgBilinear ¶

func AtgBilinear(ptr *Ctensor, input1 Ctensor, input2 Ctensor, weight Ctensor, bias Ctensor)

func AtgBinaryCrossEntropy ¶

func AtgBinaryCrossEntropy(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64)

func AtgBinaryCrossEntropyBackward ¶

func AtgBinaryCrossEntropyBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64)

func AtgBinaryCrossEntropyBackwardGradInput ¶ added in v0.4.0

func AtgBinaryCrossEntropyBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64)

func AtgBinaryCrossEntropyOut ¶

func AtgBinaryCrossEntropyOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64)

func AtgBinaryCrossEntropyWithLogits ¶

func AtgBinaryCrossEntropyWithLogits(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, posWeight Ctensor, reduction int64)

func AtgBinaryCrossEntropyWithLogitsOut ¶ added in v0.8.0

func AtgBinaryCrossEntropyWithLogitsOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, weight Ctensor, posWeight Ctensor, reduction int64)

func AtgBincount ¶

func AtgBincount(ptr *Ctensor, self Ctensor, weights Ctensor, minlength int64)

func AtgBincountOut ¶ added in v0.8.0

func AtgBincountOut(ptr *Ctensor, out Ctensor, self Ctensor, weights Ctensor, minlength int64)

func AtgBinomial ¶ added in v0.3.0

func AtgBinomial(ptr *Ctensor, count Ctensor, prob Ctensor)

func AtgBinomialOut ¶ added in v0.8.0

func AtgBinomialOut(ptr *Ctensor, out Ctensor, count Ctensor, prob Ctensor)

func AtgBitwiseAnd ¶

func AtgBitwiseAnd(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseAndScalarOut ¶ added in v0.4.0

func AtgBitwiseAndScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseAndScalarTensor ¶ added in v0.8.0

func AtgBitwiseAndScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseAndScalarTensorOut ¶ added in v0.8.0

func AtgBitwiseAndScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseAndTensor ¶ added in v0.4.0

func AtgBitwiseAndTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseAndTensorOut ¶ added in v0.4.0

func AtgBitwiseAndTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseAndTensor_ ¶ added in v0.4.0

func AtgBitwiseAndTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseAnd_ ¶

func AtgBitwiseAnd_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseLeftShift ¶ added in v0.5.0

func AtgBitwiseLeftShift(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseLeftShiftScalarTensor ¶ added in v0.5.0

func AtgBitwiseLeftShiftScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseLeftShiftScalarTensorOut ¶ added in v0.8.0

func AtgBitwiseLeftShiftScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseLeftShiftTensorOut ¶ added in v0.5.0

func AtgBitwiseLeftShiftTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseLeftShiftTensorScalar ¶ added in v0.5.0

func AtgBitwiseLeftShiftTensorScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseLeftShiftTensorScalarOut ¶ added in v0.5.0

func AtgBitwiseLeftShiftTensorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseLeftShiftTensorScalar_ ¶ added in v0.5.0

func AtgBitwiseLeftShiftTensorScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseLeftShift_ ¶ added in v0.5.0

func AtgBitwiseLeftShift_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseNot ¶

func AtgBitwiseNot(ptr *Ctensor, self Ctensor)

func AtgBitwiseNotOut ¶

func AtgBitwiseNotOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgBitwiseNot_ ¶

func AtgBitwiseNot_(ptr *Ctensor, self Ctensor)

func AtgBitwiseOr ¶

func AtgBitwiseOr(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseOrScalarOut ¶ added in v0.4.0

func AtgBitwiseOrScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseOrScalarTensor ¶ added in v0.8.0

func AtgBitwiseOrScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseOrScalarTensorOut ¶ added in v0.8.0

func AtgBitwiseOrScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseOrTensor ¶ added in v0.4.0

func AtgBitwiseOrTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseOrTensorOut ¶ added in v0.4.0

func AtgBitwiseOrTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseOrTensor_ ¶ added in v0.4.0

func AtgBitwiseOrTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseOr_ ¶

func AtgBitwiseOr_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseRightShift ¶ added in v0.5.0

func AtgBitwiseRightShift(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseRightShiftScalarTensor ¶ added in v0.5.0

func AtgBitwiseRightShiftScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseRightShiftScalarTensorOut ¶ added in v0.8.0

func AtgBitwiseRightShiftScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseRightShiftTensorOut ¶ added in v0.5.0

func AtgBitwiseRightShiftTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseRightShiftTensorScalar ¶ added in v0.5.0

func AtgBitwiseRightShiftTensorScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseRightShiftTensorScalarOut ¶ added in v0.5.0

func AtgBitwiseRightShiftTensorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseRightShiftTensorScalar_ ¶ added in v0.5.0

func AtgBitwiseRightShiftTensorScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseRightShift_ ¶ added in v0.5.0

func AtgBitwiseRightShift_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseXor ¶

func AtgBitwiseXor(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseXorScalarOut ¶ added in v0.4.0

func AtgBitwiseXorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgBitwiseXorScalarTensor ¶ added in v0.8.0

func AtgBitwiseXorScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseXorScalarTensorOut ¶ added in v0.8.0

func AtgBitwiseXorScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgBitwiseXorTensor ¶ added in v0.4.0

func AtgBitwiseXorTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseXorTensorOut ¶ added in v0.4.0

func AtgBitwiseXorTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseXorTensor_ ¶ added in v0.4.0

func AtgBitwiseXorTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgBitwiseXor_ ¶

func AtgBitwiseXor_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgBlackmanWindow ¶

func AtgBlackmanWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)

func AtgBlackmanWindowOut ¶ added in v0.8.0

func AtgBlackmanWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)

func AtgBlackmanWindowPeriodic ¶ added in v0.4.0

func AtgBlackmanWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, optionsDevice int32)

func AtgBlackmanWindowPeriodicOut ¶ added in v0.8.0

func AtgBlackmanWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)

func AtgBlockDiag ¶ added in v0.3.0

func AtgBlockDiag(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgBlockDiagOut ¶ added in v0.8.0

func AtgBlockDiagOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgBmm ¶

func AtgBmm(ptr *Ctensor, self Ctensor, mat2 Ctensor)

func AtgBmmOut ¶

func AtgBmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor)

func AtgBroadcastTo ¶ added in v0.4.0

func AtgBroadcastTo(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgBucketize ¶ added in v0.3.0

func AtgBucketize(ptr *Ctensor, self Ctensor, boundaries Ctensor, outInt32 int32, right int32)

func AtgBucketizeScalar ¶ added in v0.4.0

func AtgBucketizeScalar(ptr *Ctensor, selfScalar Cscalar, boundaries Ctensor, outInt32 int32, right int32)

func AtgBucketizeScalarOut ¶ added in v0.8.0

func AtgBucketizeScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, boundaries Ctensor, outInt32 int32, right int32)

func AtgBucketizeTensorOut ¶ added in v0.4.0

func AtgBucketizeTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, boundaries Ctensor, outInt32 int32, right int32)

func AtgCanCast ¶ added in v0.4.0

func AtgCanCast(from int32, to int32) bool

func AtgCartesianProd ¶

func AtgCartesianProd(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgCat ¶

func AtgCat(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func AtgCatOut ¶

func AtgCatOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func AtgCauchy ¶ added in v0.8.0

func AtgCauchy(ptr *Ctensor, self Ctensor, median float64, sigma float64)

func AtgCauchyOut ¶ added in v0.8.0

func AtgCauchyOut(ptr *Ctensor, out Ctensor, self Ctensor, median float64, sigma float64)

func AtgCauchy_ ¶

func AtgCauchy_(ptr *Ctensor, self Ctensor, median float64, sigma float64)

func AtgCcolIndices ¶ added in v0.8.0

func AtgCcolIndices(ptr *Ctensor, self Ctensor)

func AtgCcolIndicesCopy ¶ added in v0.8.0

func AtgCcolIndicesCopy(ptr *Ctensor, self Ctensor)

func AtgCcolIndicesCopyOut ¶ added in v0.8.0

func AtgCcolIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgCdist ¶

func AtgCdist(ptr *Ctensor, x1 Ctensor, x2 Ctensor, p float64, computeModeVal int64, computeModeNull int)

func AtgCeil ¶

func AtgCeil(ptr *Ctensor, self Ctensor)

func AtgCeilOut ¶

func AtgCeilOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgCeil_ ¶

func AtgCeil_(ptr *Ctensor, self Ctensor)

func AtgCelu ¶

func AtgCelu(ptr *Ctensor, self Ctensor)

func AtgCeluOut ¶ added in v0.8.0

func AtgCeluOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgCelu_ ¶

func AtgCelu_(ptr *Ctensor, self Ctensor)

func AtgChainMatmul ¶

func AtgChainMatmul(ptr *Ctensor, matricesData []Ctensor, matricesLen int)

func AtgChainMatmulOut ¶ added in v0.4.0

func AtgChainMatmulOut(ptr *Ctensor, out Ctensor, matricesData []Ctensor, matricesLen int)

func AtgChalf ¶ added in v0.8.0

func AtgChalf(ptr *Ctensor, self Ctensor)

func AtgChannelShuffle ¶ added in v0.3.0

func AtgChannelShuffle(ptr *Ctensor, self Ctensor, groups int64)

func AtgChannelShuffleOut ¶ added in v0.8.0

func AtgChannelShuffleOut(ptr *Ctensor, out Ctensor, self Ctensor, groups int64)

func AtgCholesky ¶

func AtgCholesky(ptr *Ctensor, self Ctensor, upper int32)

func AtgCholeskyInverse ¶

func AtgCholeskyInverse(ptr *Ctensor, self Ctensor, upper int32)

func AtgCholeskyInverseOut ¶

func AtgCholeskyInverseOut(ptr *Ctensor, out Ctensor, self Ctensor, upper int32)

func AtgCholeskyOut ¶

func AtgCholeskyOut(ptr *Ctensor, out Ctensor, self Ctensor, upper int32)

func AtgCholeskySolve ¶

func AtgCholeskySolve(ptr *Ctensor, self Ctensor, input2 Ctensor, upper int32)

func AtgCholeskySolveOut ¶

func AtgCholeskySolveOut(ptr *Ctensor, out Ctensor, self Ctensor, input2 Ctensor, upper int32)

func AtgChooseQparamsOptimized ¶ added in v0.4.0

func AtgChooseQparamsOptimized(ptr *Ctensor, input Ctensor, numel int64, nBins int64, ratio float64, bitWidth int64)

func AtgClamp ¶

func AtgClamp(ptr *Ctensor, self Ctensor, min Cscalar, max Cscalar)

func AtgClampMax ¶

func AtgClampMax(ptr *Ctensor, self Ctensor, max Cscalar)

func AtgClampMaxOut ¶

func AtgClampMaxOut(ptr *Ctensor, out Ctensor, self Ctensor, max Cscalar)

func AtgClampMaxTensor ¶ added in v0.4.0

func AtgClampMaxTensor(ptr *Ctensor, self Ctensor, max Ctensor)

func AtgClampMaxTensorOut ¶ added in v0.4.0

func AtgClampMaxTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, max Ctensor)

func AtgClampMaxTensor_ ¶ added in v0.4.0

func AtgClampMaxTensor_(ptr *Ctensor, self Ctensor, max Ctensor)

func AtgClampMax_ ¶

func AtgClampMax_(ptr *Ctensor, self Ctensor, max Cscalar)

func AtgClampMin ¶

func AtgClampMin(ptr *Ctensor, self Ctensor, min Cscalar)

func AtgClampMinOut ¶

func AtgClampMinOut(ptr *Ctensor, out Ctensor, self Ctensor, min Cscalar)

func AtgClampMinTensor ¶ added in v0.4.0

func AtgClampMinTensor(ptr *Ctensor, self Ctensor, min Ctensor)

func AtgClampMinTensorOut ¶ added in v0.4.0

func AtgClampMinTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, min Ctensor)

func AtgClampMinTensor_ ¶ added in v0.4.0

func AtgClampMinTensor_(ptr *Ctensor, self Ctensor, min Ctensor)

func AtgClampMin_ ¶

func AtgClampMin_(ptr *Ctensor, self Ctensor, min Cscalar)

func AtgClampOut ¶

func AtgClampOut(ptr *Ctensor, out Ctensor, self Ctensor, min Cscalar, max Cscalar)

func AtgClampTensor ¶ added in v0.4.0

func AtgClampTensor(ptr *Ctensor, self Ctensor, min Ctensor, max Ctensor)

func AtgClampTensorOut ¶ added in v0.4.0

func AtgClampTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, min Ctensor, max Ctensor)

func AtgClampTensor_ ¶ added in v0.4.0

func AtgClampTensor_(ptr *Ctensor, self Ctensor, min Ctensor, max Ctensor)

func AtgClamp_ ¶

func AtgClamp_(ptr *Ctensor, self Ctensor, min Cscalar, max Cscalar)

func AtgClip ¶ added in v0.3.0

func AtgClip(ptr *Ctensor, self Ctensor, min Cscalar, max Cscalar)

func AtgClipOut ¶ added in v0.3.0

func AtgClipOut(ptr *Ctensor, out Ctensor, self Ctensor, min Cscalar, max Cscalar)

func AtgClipTensor ¶ added in v0.4.0

func AtgClipTensor(ptr *Ctensor, self Ctensor, min Ctensor, max Ctensor)

func AtgClipTensorOut ¶ added in v0.4.0

func AtgClipTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, min Ctensor, max Ctensor)

func AtgClipTensor_ ¶ added in v0.4.0

func AtgClipTensor_(ptr *Ctensor, self Ctensor, min Ctensor, max Ctensor)

func AtgClip_ ¶ added in v0.3.0

func AtgClip_(ptr *Ctensor, self Ctensor, min Cscalar, max Cscalar)

func AtgClone ¶ added in v0.8.0

func AtgClone(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgCoalesce ¶

func AtgCoalesce(ptr *Ctensor, self Ctensor)

func AtgCol2im ¶

func AtgCol2im(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, kernelSizeData []int64, kernelSizeLen int, dilationData []int64, dilationLen int, paddingData []int64, paddingLen int, strideData []int64, strideLen int)

func AtgCol2imOut ¶

func AtgCol2imOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, kernelSizeData []int64, kernelSizeLen int, dilationData []int64, dilationLen int, paddingData []int64, paddingLen int, strideData []int64, strideLen int)

func AtgColIndices ¶ added in v0.4.0

func AtgColIndices(ptr *Ctensor, self Ctensor)

func AtgColIndicesCopy ¶ added in v0.8.0

func AtgColIndicesCopy(ptr *Ctensor, self Ctensor)

func AtgColIndicesCopyOut ¶ added in v0.8.0

func AtgColIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgColumnStack ¶ added in v0.4.0

func AtgColumnStack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgColumnStackOut ¶ added in v0.4.0

func AtgColumnStackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgCombinations ¶

func AtgCombinations(ptr *Ctensor, self Ctensor, r int64, withReplacement int32)

func AtgComplex ¶ added in v0.3.0

func AtgComplex(ptr *Ctensor, real Ctensor, imag Ctensor)

func AtgComplexOut ¶ added in v0.3.0

func AtgComplexOut(ptr *Ctensor, out Ctensor, real Ctensor, imag Ctensor)

func AtgConcat ¶ added in v0.5.0

func AtgConcat(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func AtgConcatOut ¶ added in v0.5.0

func AtgConcatOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func AtgConcatenate ¶ added in v0.8.0

func AtgConcatenate(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func AtgConcatenateOut ¶ added in v0.8.0

func AtgConcatenateOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func AtgConj ¶

func AtgConj(ptr *Ctensor, self Ctensor)

func AtgConjPhysical ¶ added in v0.5.0

func AtgConjPhysical(ptr *Ctensor, self Ctensor)

func AtgConjPhysicalOut ¶ added in v0.5.0

func AtgConjPhysicalOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgConjPhysical_ ¶ added in v0.5.0

func AtgConjPhysical_(ptr *Ctensor, self Ctensor)

func AtgConstantPadNd ¶

func AtgConstantPadNd(ptr *Ctensor, self Ctensor, padData []int64, padLen int)

func AtgConstantPadNdOut ¶ added in v0.8.0

func AtgConstantPadNdOut(ptr *Ctensor, out Ctensor, self Ctensor, padData []int64, padLen int)

func AtgContiguous ¶

func AtgContiguous(ptr *Ctensor, self Ctensor)

func AtgConv1d ¶

func AtgConv1d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, groups int64)

func AtgConv1dPadding ¶ added in v0.4.0

func AtgConv1dPadding(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, padding string, dilationData []int64, dilationLen int, groups int64)

func AtgConv2d ¶

func AtgConv2d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, groups int64)

func AtgConv2dPadding ¶ added in v0.4.0

func AtgConv2dPadding(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, padding string, dilationData []int64, dilationLen int, groups int64)

func AtgConv3d ¶

func AtgConv3d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, groups int64)

func AtgConv3dPadding ¶ added in v0.4.0

func AtgConv3dPadding(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, padding string, dilationData []int64, dilationLen int, groups int64)

func AtgConvDepthwise3d ¶ added in v0.4.0

func AtgConvDepthwise3d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int)

func AtgConvDepthwise3dOut ¶ added in v0.8.0

func AtgConvDepthwise3dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int)

func AtgConvTbc ¶

func AtgConvTbc(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, pad int64)

func AtgConvTbcBackward ¶

func AtgConvTbcBackward(ptr *Ctensor, self Ctensor, input Ctensor, weight Ctensor, bias Ctensor, pad int64)

func AtgConvTbcOut ¶ added in v0.8.0

func AtgConvTbcOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, pad int64)

func AtgConvTranspose1d ¶

func AtgConvTranspose1d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, groups int64, dilationData []int64, dilationLen int)

func AtgConvTranspose2d ¶

func AtgConvTranspose2d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, groups int64, dilationData []int64, dilationLen int)

func AtgConvTranspose3d ¶

func AtgConvTranspose3d(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, groups int64, dilationData []int64, dilationLen int)

func AtgConvolution ¶

func AtgConvolution(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, transposed int32, outputPaddingData []int64, outputPaddingLen int, groups int64)

func AtgConvolutionOut ¶ added in v0.8.0

func AtgConvolutionOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, transposed int32, outputPaddingData []int64, outputPaddingLen int, groups int64)

func AtgConvolutionOverrideable ¶

func AtgConvolutionOverrideable(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, transposed int32, outputPaddingData []int64, outputPaddingLen int, groups int64)

func AtgConvolutionOverrideableOut ¶ added in v0.8.0

func AtgConvolutionOverrideableOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, transposed int32, outputPaddingData []int64, outputPaddingLen int, groups int64)

func AtgCopy ¶ added in v0.8.0

func AtgCopy(ptr *Ctensor, self Ctensor, src Ctensor, nonBlocking int32)

func AtgCopyOut ¶ added in v0.8.0

func AtgCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, nonBlocking int32)

func AtgCopySparseToSparse ¶ added in v0.8.0

func AtgCopySparseToSparse(ptr *Ctensor, self Ctensor, src Ctensor, nonBlocking int32)

func AtgCopySparseToSparseOut ¶ added in v0.8.0

func AtgCopySparseToSparseOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, nonBlocking int32)

func AtgCopySparseToSparse_ ¶

func AtgCopySparseToSparse_(ptr *Ctensor, self Ctensor, src Ctensor, nonBlocking int32)

func AtgCopysign ¶ added in v0.4.0

func AtgCopysign(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgCopysignOut ¶ added in v0.4.0

func AtgCopysignOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgCopysignScalar ¶ added in v0.4.0

func AtgCopysignScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgCopysignScalarOut ¶ added in v0.4.0

func AtgCopysignScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgCopysignScalar_ ¶ added in v0.4.0

func AtgCopysignScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgCopysign_ ¶ added in v0.4.0

func AtgCopysign_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgCorrcoef ¶ added in v0.5.0

func AtgCorrcoef(ptr *Ctensor, self Ctensor)

func AtgCos ¶

func AtgCos(ptr *Ctensor, self Ctensor)

func AtgCosOut ¶

func AtgCosOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgCos_ ¶

func AtgCos_(ptr *Ctensor, self Ctensor)

func AtgCosh ¶

func AtgCosh(ptr *Ctensor, self Ctensor)

func AtgCoshOut ¶

func AtgCoshOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgCosh_ ¶

func AtgCosh_(ptr *Ctensor, self Ctensor)

func AtgCosineEmbeddingLoss ¶

func AtgCosineEmbeddingLoss(ptr *Ctensor, input1 Ctensor, input2 Ctensor, target Ctensor, margin float64, reduction int64)

func AtgCosineSimilarity ¶

func AtgCosineSimilarity(ptr *Ctensor, x1 Ctensor, x2 Ctensor, dim int64, eps float64)

func AtgCountNonzero ¶ added in v0.3.0

func AtgCountNonzero(ptr *Ctensor, self Ctensor, dimVal int64, dimNull int)

func AtgCountNonzeroDimIntlist ¶ added in v0.4.0

func AtgCountNonzeroDimIntlist(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)

func AtgCountNonzeroDimIntlistOut ¶ added in v0.8.0

func AtgCountNonzeroDimIntlistOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int)

func AtgCountNonzeroOut ¶ added in v0.8.0

func AtgCountNonzeroOut(ptr *Ctensor, out Ctensor, self Ctensor, dimVal int64, dimNull int)

func AtgCov ¶ added in v0.5.0

func AtgCov(ptr *Ctensor, self Ctensor, correction int64, fweights Ctensor, aweights Ctensor)

func AtgCross ¶

func AtgCross(ptr *Ctensor, self Ctensor, other Ctensor, dimVal int64, dimNull int)

func AtgCrossEntropyLoss ¶ added in v0.4.0

func AtgCrossEntropyLoss(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64, labelSmoothing float64)

func AtgCrossOut ¶

func AtgCrossOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, dimVal int64, dimNull int)

func AtgCrowIndices ¶ added in v0.4.0

func AtgCrowIndices(ptr *Ctensor, self Ctensor)

func AtgCrowIndicesCopy ¶ added in v0.8.0

func AtgCrowIndicesCopy(ptr *Ctensor, self Ctensor)

func AtgCrowIndicesCopyOut ¶ added in v0.8.0

func AtgCrowIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgCtcLoss ¶

func AtgCtcLoss(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, blank int64, reduction int64, zeroInfinity int32)

func AtgCtcLossTensor ¶ added in v0.4.0

func AtgCtcLossTensor(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengths Ctensor, targetLengths Ctensor, blank int64, reduction int64, zeroInfinity int32)

func AtgCudnnAffineGridGenerator ¶

func AtgCudnnAffineGridGenerator(ptr *Ctensor, theta Ctensor, n int64, c int64, h int64, w int64)

func AtgCudnnAffineGridGeneratorBackward ¶

func AtgCudnnAffineGridGeneratorBackward(ptr *Ctensor, grad Ctensor, n int64, c int64, h int64, w int64)

func AtgCudnnAffineGridGeneratorBackwardOut ¶ added in v0.8.0

func AtgCudnnAffineGridGeneratorBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, n int64, c int64, h int64, w int64)

func AtgCudnnAffineGridGeneratorOut ¶ added in v0.8.0

func AtgCudnnAffineGridGeneratorOut(ptr *Ctensor, out Ctensor, theta Ctensor, n int64, c int64, h int64, w int64)

func AtgCudnnBatchNorm ¶

func AtgCudnnBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, exponentialAverageFactor float64, epsilon float64)

func AtgCudnnBatchNormBackward ¶

func AtgCudnnBatchNormBackward(ptr *Ctensor, input Ctensor, gradOutput Ctensor, weight Ctensor, runningMean Ctensor, runningVar Ctensor, saveMean Ctensor, saveVar Ctensor, epsilon float64, reserveSpace Ctensor)

func AtgCudnnBatchNormBackwardOut ¶ added in v0.8.0

func AtgCudnnBatchNormBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, gradOutput Ctensor, weight Ctensor, runningMean Ctensor, runningVar Ctensor, saveMean Ctensor, saveVar Ctensor, epsilon float64, reserveSpace Ctensor)

func AtgCudnnBatchNormOut ¶ added in v0.8.0

func AtgCudnnBatchNormOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, exponentialAverageFactor float64, epsilon float64)

func AtgCudnnConvolution ¶

func AtgCudnnConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32, allowTf32 int32)

func AtgCudnnConvolutionAddRelu ¶ added in v0.4.0

func AtgCudnnConvolutionAddRelu(ptr *Ctensor, self Ctensor, weight Ctensor, z Ctensor, alpha Cscalar, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, groups int64)

func AtgCudnnConvolutionAddReluOut ¶ added in v0.8.0

func AtgCudnnConvolutionAddReluOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, z Ctensor, alpha Cscalar, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, groups int64)

func AtgCudnnConvolutionOut ¶ added in v0.8.0

func AtgCudnnConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32, allowTf32 int32)

func AtgCudnnConvolutionRelu ¶ added in v0.4.0

func AtgCudnnConvolutionRelu(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, groups int64)

func AtgCudnnConvolutionReluOut ¶ added in v0.8.0

func AtgCudnnConvolutionReluOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, groups int64)

func AtgCudnnConvolutionTranspose ¶

func AtgCudnnConvolutionTranspose(ptr *Ctensor, self Ctensor, weight Ctensor, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32, allowTf32 int32)

func AtgCudnnConvolutionTransposeOut ¶ added in v0.8.0

func AtgCudnnConvolutionTransposeOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32, allowTf32 int32)

func AtgCudnnGridSampler ¶

func AtgCudnnGridSampler(ptr *Ctensor, self Ctensor, grid Ctensor)

func AtgCudnnGridSamplerBackward ¶

func AtgCudnnGridSamplerBackward(ptr *Ctensor, self Ctensor, grid Ctensor, gradOutput Ctensor)

func AtgCudnnGridSamplerBackwardOut ¶ added in v0.8.0

func AtgCudnnGridSamplerBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, grid Ctensor, gradOutput Ctensor)

func AtgCudnnGridSamplerOut ¶ added in v0.8.0

func AtgCudnnGridSamplerOut(ptr *Ctensor, out Ctensor, self Ctensor, grid Ctensor)

func AtgCudnnIsAcceptable ¶ added in v0.4.0

func AtgCudnnIsAcceptable(self Ctensor) bool

func AtgCummax ¶

func AtgCummax(ptr *Ctensor, self Ctensor, dim int64)

func AtgCummaxOut ¶

func AtgCummaxOut(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64)

func AtgCummaxminBackward ¶ added in v0.3.0

func AtgCummaxminBackward(ptr *Ctensor, grad Ctensor, input Ctensor, indices Ctensor, dim int64)

func AtgCummin ¶

func AtgCummin(ptr *Ctensor, self Ctensor, dim int64)

func AtgCumminOut ¶

func AtgCumminOut(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64)

func AtgCumprod ¶

func AtgCumprod(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func AtgCumprodBackward ¶ added in v0.3.0

func AtgCumprodBackward(ptr *Ctensor, grad Ctensor, input Ctensor, dim int64, output Ctensor)

func AtgCumprodOut ¶

func AtgCumprodOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, dtype int32)

func AtgCumprod_ ¶ added in v0.4.0

func AtgCumprod_(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func AtgCumsum ¶

func AtgCumsum(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func AtgCumsumOut ¶

func AtgCumsumOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, dtype int32)

func AtgCumsum_ ¶ added in v0.4.0

func AtgCumsum_(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func AtgCumulativeTrapezoid ¶ added in v0.5.0

func AtgCumulativeTrapezoid(ptr *Ctensor, y Ctensor, dim int64)

func AtgCumulativeTrapezoidX ¶ added in v0.5.0

func AtgCumulativeTrapezoidX(ptr *Ctensor, y Ctensor, x Ctensor, dim int64)

func AtgData ¶

func AtgData(ptr *Ctensor, self Ctensor)

func AtgDeg2rad ¶ added in v0.3.0

func AtgDeg2rad(ptr *Ctensor, self Ctensor)

func AtgDeg2radOut ¶ added in v0.3.0

func AtgDeg2radOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgDeg2rad_ ¶ added in v0.3.0

func AtgDeg2rad_(ptr *Ctensor, self Ctensor)

func AtgDenseDim ¶ added in v0.4.0

func AtgDenseDim(self Ctensor) int64

func AtgDequantize ¶

func AtgDequantize(ptr *Ctensor, self Ctensor)

func AtgDet ¶

func AtgDet(ptr *Ctensor, self Ctensor)

func AtgDetach ¶

func AtgDetach(ptr *Ctensor, self Ctensor)

func AtgDetachCopy ¶ added in v0.8.0

func AtgDetachCopy(ptr *Ctensor, self Ctensor)

func AtgDetachCopyOut ¶ added in v0.8.0

func AtgDetachCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgDetach_ ¶

func AtgDetach_(ptr *Ctensor, self Ctensor)

func AtgDiag ¶

func AtgDiag(ptr *Ctensor, self Ctensor, diagonal int64)

func AtgDiagEmbed ¶

func AtgDiagEmbed(ptr *Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)

func AtgDiagEmbedOut ¶ added in v0.8.0

func AtgDiagEmbedOut(ptr *Ctensor, out Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)

func AtgDiagOut ¶

func AtgDiagOut(ptr *Ctensor, out Ctensor, self Ctensor, diagonal int64)

func AtgDiagflat ¶

func AtgDiagflat(ptr *Ctensor, self Ctensor, offset int64)

func AtgDiagonal ¶

func AtgDiagonal(ptr *Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)

func AtgDiagonalBackward ¶ added in v0.3.0

func AtgDiagonalBackward(ptr *Ctensor, gradOutput Ctensor, inputSizesData []int64, inputSizesLen int, offset int64, dim1 int64, dim2 int64)

func AtgDiagonalBackwardOut ¶ added in v0.8.0

func AtgDiagonalBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, inputSizesData []int64, inputSizesLen int, offset int64, dim1 int64, dim2 int64)

func AtgDiagonalCopy ¶ added in v0.8.0

func AtgDiagonalCopy(ptr *Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)

func AtgDiagonalCopyOut ¶ added in v0.8.0

func AtgDiagonalCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, offset int64, dim1 int64, dim2 int64)

func AtgDiagonalScatter ¶ added in v0.7.0

func AtgDiagonalScatter(ptr *Ctensor, self Ctensor, src Ctensor, offset int64, dim1 int64, dim2 int64)

func AtgDiagonalScatterOut ¶ added in v0.8.0

func AtgDiagonalScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, offset int64, dim1 int64, dim2 int64)

func AtgDiff ¶ added in v0.4.0

func AtgDiff(ptr *Ctensor, self Ctensor, n int64, dim int64, prepend Ctensor, append Ctensor)

func AtgDiffOut ¶ added in v0.4.0

func AtgDiffOut(ptr *Ctensor, out Ctensor, self Ctensor, n int64, dim int64, prepend Ctensor, append Ctensor)

func AtgDigamma ¶

func AtgDigamma(ptr *Ctensor, self Ctensor)

func AtgDigammaOut ¶

func AtgDigammaOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgDigamma_ ¶

func AtgDigamma_(ptr *Ctensor, self Ctensor)

func AtgDist ¶

func AtgDist(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgDistOut ¶ added in v0.8.0

func AtgDistOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgDiv ¶

func AtgDiv(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgDivOut ¶

func AtgDivOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgDivOutMode ¶ added in v0.4.0

func AtgDivOutMode(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, roundingMode string)

func AtgDivScalar ¶ added in v0.4.0

func AtgDivScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgDivScalarMode ¶ added in v0.4.0

func AtgDivScalarMode(ptr *Ctensor, self Ctensor, other Cscalar, roundingMode string)

func AtgDivScalarModeOut ¶ added in v0.8.0

func AtgDivScalarModeOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar, roundingMode string)

func AtgDivScalarMode_ ¶ added in v0.4.0

func AtgDivScalarMode_(ptr *Ctensor, self Ctensor, other Cscalar, roundingMode string)

func AtgDivScalarOut ¶ added in v0.8.0

func AtgDivScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgDivScalar_ ¶ added in v0.4.0

func AtgDivScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgDivTensorMode ¶ added in v0.4.0

func AtgDivTensorMode(ptr *Ctensor, self Ctensor, other Ctensor, roundingMode string)

func AtgDivTensorMode_ ¶ added in v0.4.0

func AtgDivTensorMode_(ptr *Ctensor, self Ctensor, other Ctensor, roundingMode string)

func AtgDiv_ ¶

func AtgDiv_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgDivide ¶ added in v0.3.0

func AtgDivide(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgDivideOut ¶ added in v0.3.0

func AtgDivideOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgDivideOutMode ¶ added in v0.4.0

func AtgDivideOutMode(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, roundingMode string)

func AtgDivideScalar ¶ added in v0.4.0

func AtgDivideScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgDivideScalarMode ¶ added in v0.4.0

func AtgDivideScalarMode(ptr *Ctensor, self Ctensor, other Cscalar, roundingMode string)

func AtgDivideScalarMode_ ¶ added in v0.4.0

func AtgDivideScalarMode_(ptr *Ctensor, self Ctensor, other Cscalar, roundingMode string)

func AtgDivideScalar_ ¶ added in v0.4.0

func AtgDivideScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgDivideTensorMode ¶ added in v0.4.0

func AtgDivideTensorMode(ptr *Ctensor, self Ctensor, other Ctensor, roundingMode string)

func AtgDivideTensorMode_ ¶ added in v0.4.0

func AtgDivideTensorMode_(ptr *Ctensor, self Ctensor, other Ctensor, roundingMode string)

func AtgDivide_ ¶ added in v0.3.0

func AtgDivide_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgDot ¶

func AtgDot(ptr *Ctensor, self Ctensor, tensor Ctensor)

func AtgDotOut ¶

func AtgDotOut(ptr *Ctensor, out Ctensor, self Ctensor, tensor Ctensor)

func AtgDropout ¶

func AtgDropout(ptr *Ctensor, input Ctensor, p float64, train int32)

func AtgDropout_ ¶

func AtgDropout_(ptr *Ctensor, self Ctensor, p float64, train int32)

func AtgDstack ¶ added in v0.3.0

func AtgDstack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgDstackOut ¶ added in v0.3.0

func AtgDstackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgEinsum ¶

func AtgEinsum(ptr *Ctensor, equation string, tensorsData []Ctensor, tensorsLen int, pathData []int64, pathLen int)

func AtgElu ¶

func AtgElu(ptr *Ctensor, self Ctensor)

func AtgEluBackward ¶

func AtgEluBackward(ptr *Ctensor, gradOutput Ctensor, alpha Cscalar, scale Cscalar, inputScale Cscalar, isResult int32, selfOrResult Ctensor)

func AtgEluBackwardGradInput ¶ added in v0.5.0

func AtgEluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, alpha Cscalar, scale Cscalar, inputScale Cscalar, isResult int32, selfOrResult Ctensor)

func AtgEluOut ¶

func AtgEluOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgElu_ ¶

func AtgElu_(ptr *Ctensor, self Ctensor)

func AtgEmbedding ¶

func AtgEmbedding(ptr *Ctensor, weight Ctensor, indices Ctensor, paddingIdx int64, scaleGradByFreq int32, sparse int32)

func AtgEmbeddingBackward ¶

func AtgEmbeddingBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, numWeights int64, paddingIdx int64, scaleGradByFreq int32, sparse int32)

func AtgEmbeddingBag ¶

func AtgEmbeddingBag(ptr *Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, scaleGradByFreq int32, mode int64, sparse int32, perSampleWeights Ctensor, includeLastOffset int32)

func AtgEmbeddingBagPaddingIdx ¶ added in v0.4.0

func AtgEmbeddingBagPaddingIdx(ptr *Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, scaleGradByFreq int32, mode int64, sparse int32, perSampleWeights Ctensor, includeLastOffset int32, paddingIdxVal int64, paddingIdxNull int)

func AtgEmbeddingDenseBackward ¶

func AtgEmbeddingDenseBackward(ptr *Ctensor, gradOutput Ctensor, indices Ctensor, numWeights int64, paddingIdx int64, scaleGradByFreq int32)

func AtgEmbeddingDenseBackwardOut ¶ added in v0.8.0

func AtgEmbeddingDenseBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, indices Ctensor, numWeights int64, paddingIdx int64, scaleGradByFreq int32)

func AtgEmbeddingOut ¶ added in v0.8.0

func AtgEmbeddingOut(ptr *Ctensor, out Ctensor, weight Ctensor, indices Ctensor, paddingIdx int64, scaleGradByFreq int32, sparse int32)

func AtgEmbeddingRenorm ¶ added in v0.8.0

func AtgEmbeddingRenorm(ptr *Ctensor, self Ctensor, indices Ctensor, maxNorm float64, normType float64)

func AtgEmbeddingRenormOut ¶ added in v0.8.0

func AtgEmbeddingRenormOut(ptr *Ctensor, out Ctensor, self Ctensor, indices Ctensor, maxNorm float64, normType float64)

func AtgEmbeddingRenorm_ ¶

func AtgEmbeddingRenorm_(ptr *Ctensor, self Ctensor, indices Ctensor, maxNorm float64, normType float64)

func AtgEmbeddingSparseBackward ¶

func AtgEmbeddingSparseBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, numWeights int64, paddingIdx int64, scaleGradByFreq int32)

func AtgEmpty ¶

func AtgEmpty(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgEmptyLike ¶

func AtgEmptyLike(ptr *Ctensor, self Ctensor)

func AtgEmptyLikeOut ¶ added in v0.8.0

func AtgEmptyLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgEmptyOut ¶

func AtgEmptyOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)

func AtgEmptyPermuted ¶ added in v0.9.0

func AtgEmptyPermuted(ptr *Ctensor, sizeData []int64, sizeLen int, physicalLayoutData []int64, physicalLayoutLen int, optionsKind int32, optionsDevice int32)

func AtgEmptyPermutedOut ¶ added in v0.9.0

func AtgEmptyPermutedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, physicalLayoutData []int64, physicalLayoutLen int)

func AtgEmptyQuantized ¶ added in v0.3.0

func AtgEmptyQuantized(ptr *Ctensor, sizeData []int64, sizeLen int, qtensor Ctensor, optionsKind int32, optionsDevice int32)

func AtgEmptyQuantizedOut ¶ added in v0.8.0

func AtgEmptyQuantizedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, qtensor Ctensor)

func AtgEmptyStrided ¶

func AtgEmptyStrided(ptr *Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int, optionsKind int32, optionsDevice int32)

func AtgEmptyStridedOut ¶ added in v0.8.0

func AtgEmptyStridedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int)

func AtgEq ¶

func AtgEq(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgEqScalarOut ¶ added in v0.4.0

func AtgEqScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgEqTensor ¶ added in v0.4.0

func AtgEqTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgEqTensorOut ¶ added in v0.4.0

func AtgEqTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgEqTensor_ ¶ added in v0.4.0

func AtgEqTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgEq_ ¶

func AtgEq_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgEqual ¶ added in v0.4.0

func AtgEqual(self Ctensor, other Ctensor) bool

func AtgErf ¶

func AtgErf(ptr *Ctensor, self Ctensor)

func AtgErfOut ¶

func AtgErfOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgErf_ ¶

func AtgErf_(ptr *Ctensor, self Ctensor)

func AtgErfc ¶

func AtgErfc(ptr *Ctensor, self Ctensor)

func AtgErfcOut ¶

func AtgErfcOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgErfc_ ¶

func AtgErfc_(ptr *Ctensor, self Ctensor)

func AtgErfinv ¶

func AtgErfinv(ptr *Ctensor, self Ctensor)

func AtgErfinvOut ¶

func AtgErfinvOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgErfinv_ ¶

func AtgErfinv_(ptr *Ctensor, self Ctensor)

func AtgExp ¶

func AtgExp(ptr *Ctensor, self Ctensor)

func AtgExp2 ¶ added in v0.3.0

func AtgExp2(ptr *Ctensor, self Ctensor)

func AtgExp2Out ¶ added in v0.3.0

func AtgExp2Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgExp2_ ¶ added in v0.3.0

func AtgExp2_(ptr *Ctensor, self Ctensor)

func AtgExpOut ¶

func AtgExpOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgExp_ ¶

func AtgExp_(ptr *Ctensor, self Ctensor)

func AtgExpand ¶

func AtgExpand(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, implicit int32)

func AtgExpandAs ¶

func AtgExpandAs(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgExpandCopy ¶ added in v0.8.0

func AtgExpandCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, implicit int32)

func AtgExpandCopyOut ¶ added in v0.8.0

func AtgExpandCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, implicit int32)

func AtgExpm1 ¶

func AtgExpm1(ptr *Ctensor, self Ctensor)

func AtgExpm1Out ¶

func AtgExpm1Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgExpm1_ ¶

func AtgExpm1_(ptr *Ctensor, self Ctensor)

func AtgExponential ¶ added in v0.8.0

func AtgExponential(ptr *Ctensor, self Ctensor, lambd float64)

func AtgExponentialOut ¶ added in v0.8.0

func AtgExponentialOut(ptr *Ctensor, out Ctensor, self Ctensor, lambd float64)

func AtgExponential_ ¶

func AtgExponential_(ptr *Ctensor, self Ctensor, lambd float64)

func AtgEye ¶

func AtgEye(ptr *Ctensor, n int64, optionsKind int32, optionsDevice int32)

func AtgEyeM ¶ added in v0.4.0

func AtgEyeM(ptr *Ctensor, n int64, m int64, optionsKind int32, optionsDevice int32)

func AtgEyeMOut ¶ added in v0.4.0

func AtgEyeMOut(ptr *Ctensor, out Ctensor, n int64, m int64)

func AtgEyeOut ¶

func AtgEyeOut(ptr *Ctensor, out Ctensor, n int64)

func AtgFakeQuantizePerChannelAffine ¶

func AtgFakeQuantizePerChannelAffine(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64, quantMin int64, quantMax int64)

func AtgFakeQuantizePerChannelAffineCachemask ¶ added in v0.4.0

func AtgFakeQuantizePerChannelAffineCachemask(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64, quantMin int64, quantMax int64)

func AtgFakeQuantizePerChannelAffineCachemaskBackward ¶ added in v0.4.0

func AtgFakeQuantizePerChannelAffineCachemaskBackward(ptr *Ctensor, grad Ctensor, mask Ctensor)

func AtgFakeQuantizePerChannelAffineCachemaskOut ¶ added in v0.8.0

func AtgFakeQuantizePerChannelAffineCachemaskOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64, quantMin int64, quantMax int64)

func AtgFakeQuantizePerTensorAffine ¶

func AtgFakeQuantizePerTensorAffine(ptr *Ctensor, self Ctensor, scale float64, zeroPoint int64, quantMin int64, quantMax int64)

func AtgFakeQuantizePerTensorAffineCachemask ¶ added in v0.4.0

func AtgFakeQuantizePerTensorAffineCachemask(ptr *Ctensor, self Ctensor, scale float64, zeroPoint int64, quantMin int64, quantMax int64)

func AtgFakeQuantizePerTensorAffineCachemaskBackward ¶ added in v0.4.0

func AtgFakeQuantizePerTensorAffineCachemaskBackward(ptr *Ctensor, grad Ctensor, mask Ctensor)

func AtgFakeQuantizePerTensorAffineCachemaskOut ¶ added in v0.8.0

func AtgFakeQuantizePerTensorAffineCachemaskOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, scale float64, zeroPoint int64, quantMin int64, quantMax int64)

func AtgFakeQuantizePerTensorAffineTensorQparams ¶ added in v0.5.0

func AtgFakeQuantizePerTensorAffineTensorQparams(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, quantMin int64, quantMax int64)

func AtgFbgemmLinearFp16Weight ¶

func AtgFbgemmLinearFp16Weight(ptr *Ctensor, input Ctensor, packedWeight Ctensor, bias Ctensor)

func AtgFbgemmLinearFp16WeightFp32Activation ¶

func AtgFbgemmLinearFp16WeightFp32Activation(ptr *Ctensor, input Ctensor, packedWeight Ctensor, bias Ctensor)

func AtgFbgemmLinearInt8Weight ¶

func AtgFbgemmLinearInt8Weight(ptr *Ctensor, input Ctensor, weight Ctensor, packed Ctensor, colOffsets Ctensor, weightScale Cscalar, weightZeroPoint Cscalar, bias Ctensor)

func AtgFbgemmLinearInt8WeightFp32Activation ¶

func AtgFbgemmLinearInt8WeightFp32Activation(ptr *Ctensor, input Ctensor, weight Ctensor, packed Ctensor, colOffsets Ctensor, weightScale Cscalar, weightZeroPoint Cscalar, bias Ctensor)

func AtgFbgemmPackGemmMatrixFp16 ¶

func AtgFbgemmPackGemmMatrixFp16(ptr *Ctensor, input Ctensor)

func AtgFbgemmPackQuantizedMatrix ¶

func AtgFbgemmPackQuantizedMatrix(ptr *Ctensor, input Ctensor)

func AtgFbgemmPackQuantizedMatrixKn ¶ added in v0.4.0

func AtgFbgemmPackQuantizedMatrixKn(ptr *Ctensor, input Ctensor, k int64, n int64)

func AtgFeatureAlphaDropout ¶

func AtgFeatureAlphaDropout(ptr *Ctensor, input Ctensor, p float64, train int32)

func AtgFeatureAlphaDropout_ ¶

func AtgFeatureAlphaDropout_(ptr *Ctensor, self Ctensor, p float64, train int32)

func AtgFeatureDropout ¶

func AtgFeatureDropout(ptr *Ctensor, input Ctensor, p float64, train int32)

func AtgFeatureDropout_ ¶

func AtgFeatureDropout_(ptr *Ctensor, self Ctensor, p float64, train int32)

func AtgFftFft ¶ added in v0.3.0

func AtgFftFft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftFft2 ¶ added in v0.4.0

func AtgFftFft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftFft2Out ¶ added in v0.4.0

func AtgFftFft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftFftOut ¶ added in v0.4.0

func AtgFftFftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftFftfreq ¶ added in v0.4.0

func AtgFftFftfreq(ptr *Ctensor, n int64, d float64, optionsKind int32, optionsDevice int32)

func AtgFftFftfreqOut ¶ added in v0.4.0

func AtgFftFftfreqOut(ptr *Ctensor, out Ctensor, n int64, d float64)

func AtgFftFftn ¶ added in v0.3.0

func AtgFftFftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftFftnOut ¶ added in v0.4.0

func AtgFftFftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftFftshift ¶ added in v0.4.0

func AtgFftFftshift(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)

func AtgFftHfft ¶ added in v0.3.0

func AtgFftHfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftHfft2 ¶ added in v0.7.0

func AtgFftHfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftHfft2Out ¶ added in v0.7.0

func AtgFftHfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftHfftOut ¶ added in v0.4.0

func AtgFftHfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftHfftn ¶ added in v0.7.0

func AtgFftHfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftHfftnOut ¶ added in v0.7.0

func AtgFftHfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIfft ¶ added in v0.3.0

func AtgFftIfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftIfft2 ¶ added in v0.4.0

func AtgFftIfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIfft2Out ¶ added in v0.4.0

func AtgFftIfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIfftOut ¶ added in v0.4.0

func AtgFftIfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftIfftn ¶ added in v0.3.0

func AtgFftIfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIfftnOut ¶ added in v0.4.0

func AtgFftIfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIfftshift ¶ added in v0.4.0

func AtgFftIfftshift(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)

func AtgFftIhfft ¶ added in v0.3.0

func AtgFftIhfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftIhfft2 ¶ added in v0.7.0

func AtgFftIhfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIhfft2Out ¶ added in v0.7.0

func AtgFftIhfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIhfftOut ¶ added in v0.4.0

func AtgFftIhfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftIhfftn ¶ added in v0.7.0

func AtgFftIhfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIhfftnOut ¶ added in v0.7.0

func AtgFftIhfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIrfft ¶ added in v0.3.0

func AtgFftIrfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftIrfft2 ¶ added in v0.4.0

func AtgFftIrfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIrfft2Out ¶ added in v0.4.0

func AtgFftIrfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIrfftOut ¶ added in v0.4.0

func AtgFftIrfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftIrfftn ¶ added in v0.3.0

func AtgFftIrfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftIrfftnOut ¶ added in v0.4.0

func AtgFftIrfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftRfft ¶ added in v0.3.0

func AtgFftRfft(ptr *Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftRfft2 ¶ added in v0.4.0

func AtgFftRfft2(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftRfft2Out ¶ added in v0.4.0

func AtgFftRfft2Out(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftRfftOut ¶ added in v0.4.0

func AtgFftRfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, norm string)

func AtgFftRfftfreq ¶ added in v0.4.0

func AtgFftRfftfreq(ptr *Ctensor, n int64, d float64, optionsKind int32, optionsDevice int32)

func AtgFftRfftfreqOut ¶ added in v0.4.0

func AtgFftRfftfreqOut(ptr *Ctensor, out Ctensor, n int64, d float64)

func AtgFftRfftn ¶ added in v0.3.0

func AtgFftRfftn(ptr *Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFftRfftnOut ¶ added in v0.4.0

func AtgFftRfftnOut(ptr *Ctensor, out Ctensor, self Ctensor, sData []int64, sLen int, dimData []int64, dimLen int, norm string)

func AtgFill ¶ added in v0.8.0

func AtgFill(ptr *Ctensor, self Ctensor, value Cscalar)

func AtgFillDiagonal_ ¶

func AtgFillDiagonal_(ptr *Ctensor, self Ctensor, fillValue Cscalar, wrap int32)

func AtgFillScalarOut ¶ added in v0.8.0

func AtgFillScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, value Cscalar)

func AtgFillTensor ¶ added in v0.8.0

func AtgFillTensor(ptr *Ctensor, self Ctensor, value Ctensor)

func AtgFillTensorOut ¶ added in v0.8.0

func AtgFillTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, value Ctensor)

func AtgFillTensor_ ¶ added in v0.4.0

func AtgFillTensor_(ptr *Ctensor, self Ctensor, value Ctensor)

func AtgFill_ ¶

func AtgFill_(ptr *Ctensor, self Ctensor, value Cscalar)

func AtgFix ¶ added in v0.3.0

func AtgFix(ptr *Ctensor, self Ctensor)

func AtgFixOut ¶ added in v0.3.0

func AtgFixOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgFix_ ¶ added in v0.3.0

func AtgFix_(ptr *Ctensor, self Ctensor)

func AtgFlatten ¶

func AtgFlatten(ptr *Ctensor, self Ctensor, startDim int64, endDim int64)

func AtgFlattenDenseTensors ¶ added in v0.4.0

func AtgFlattenDenseTensors(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgFlip ¶

func AtgFlip(ptr *Ctensor, self Ctensor, dimsData []int64, dimsLen int)

func AtgFlipOut ¶ added in v0.8.0

func AtgFlipOut(ptr *Ctensor, out Ctensor, self Ctensor, dimsData []int64, dimsLen int)

func AtgFliplr ¶ added in v0.3.0

func AtgFliplr(ptr *Ctensor, self Ctensor)

func AtgFlipud ¶ added in v0.3.0

func AtgFlipud(ptr *Ctensor, self Ctensor)

func AtgFloatPower ¶ added in v0.4.0

func AtgFloatPower(ptr *Ctensor, self Ctensor, exponent Ctensor)

func AtgFloatPowerScalar ¶ added in v0.4.0

func AtgFloatPowerScalar(ptr *Ctensor, selfScalar Cscalar, exponent Ctensor)

func AtgFloatPowerScalarOut ¶ added in v0.4.0

func AtgFloatPowerScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, exponent Ctensor)

func AtgFloatPowerTensorScalar ¶ added in v0.4.0

func AtgFloatPowerTensorScalar(ptr *Ctensor, self Ctensor, exponent Cscalar)

func AtgFloatPowerTensorScalarOut ¶ added in v0.4.0

func AtgFloatPowerTensorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, exponent Cscalar)

func AtgFloatPowerTensorTensorOut ¶ added in v0.4.0

func AtgFloatPowerTensorTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, exponent Ctensor)

func AtgFloatPowerTensor_ ¶ added in v0.4.0

func AtgFloatPowerTensor_(ptr *Ctensor, self Ctensor, exponent Ctensor)

func AtgFloatPower_ ¶ added in v0.4.0

func AtgFloatPower_(ptr *Ctensor, self Ctensor, exponent Cscalar)

func AtgFloor ¶

func AtgFloor(ptr *Ctensor, self Ctensor)

func AtgFloorDivide ¶

func AtgFloorDivide(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgFloorDivideOut ¶

func AtgFloorDivideOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgFloorDivideScalar ¶ added in v0.4.0

func AtgFloorDivideScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgFloorDivideScalar_ ¶ added in v0.4.0

func AtgFloorDivideScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgFloorDivide_ ¶

func AtgFloorDivide_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgFloorOut ¶

func AtgFloorOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgFloor_ ¶

func AtgFloor_(ptr *Ctensor, self Ctensor)

func AtgFmax ¶ added in v0.4.0

func AtgFmax(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgFmaxOut ¶ added in v0.4.0

func AtgFmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgFmin ¶ added in v0.4.0

func AtgFmin(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgFminOut ¶ added in v0.4.0

func AtgFminOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgFmod ¶

func AtgFmod(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgFmodScalarOut ¶ added in v0.4.0

func AtgFmodScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgFmodTensor ¶ added in v0.4.0

func AtgFmodTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgFmodTensorOut ¶ added in v0.4.0

func AtgFmodTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgFmodTensor_ ¶ added in v0.4.0

func AtgFmodTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgFmod_ ¶

func AtgFmod_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgFrac ¶

func AtgFrac(ptr *Ctensor, self Ctensor)

func AtgFracOut ¶

func AtgFracOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgFrac_ ¶

func AtgFrac_(ptr *Ctensor, self Ctensor)

func AtgFractionalMaxPool2d ¶

func AtgFractionalMaxPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, outputSizeData []int64, outputSizeLen int, randomSamples Ctensor)

func AtgFractionalMaxPool2dBackward ¶

func AtgFractionalMaxPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, outputSizeData []int64, outputSizeLen int, indices Ctensor)

func AtgFractionalMaxPool2dBackwardGradInput ¶ added in v0.4.0

func AtgFractionalMaxPool2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, outputSizeData []int64, outputSizeLen int, indices Ctensor)

func AtgFractionalMaxPool2dOutput ¶ added in v0.4.0

func AtgFractionalMaxPool2dOutput(ptr *Ctensor, output Ctensor, indices Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, outputSizeData []int64, outputSizeLen int, randomSamples Ctensor)

func AtgFractionalMaxPool3d ¶

func AtgFractionalMaxPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, outputSizeData []int64, outputSizeLen int, randomSamples Ctensor)

func AtgFractionalMaxPool3dBackward ¶

func AtgFractionalMaxPool3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, outputSizeData []int64, outputSizeLen int, indices Ctensor)

func AtgFractionalMaxPool3dBackwardGradInput ¶ added in v0.4.0

func AtgFractionalMaxPool3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, outputSizeData []int64, outputSizeLen int, indices Ctensor)

func AtgFractionalMaxPool3dOutput ¶ added in v0.4.0

func AtgFractionalMaxPool3dOutput(ptr *Ctensor, output Ctensor, indices Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, outputSizeData []int64, outputSizeLen int, randomSamples Ctensor)

func AtgFrexp ¶ added in v0.4.0

func AtgFrexp(ptr *Ctensor, self Ctensor)

func AtgFrexpTensorOut ¶ added in v0.4.0

func AtgFrexpTensorOut(ptr *Ctensor, mantissa Ctensor, exponent Ctensor, self Ctensor)

func AtgFrobeniusNorm ¶

func AtgFrobeniusNorm(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgFrobeniusNormOut ¶

func AtgFrobeniusNormOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgFromFile ¶

func AtgFromFile(ptr *Ctensor, filename string, shared int32, sizeVal int64, sizeNull int, optionsKind int32, optionsDevice int32)

func AtgFromFileOut ¶ added in v0.8.0

func AtgFromFileOut(ptr *Ctensor, out Ctensor, filename string, shared int32, sizeVal int64, sizeNull int)

func AtgFull ¶

func AtgFull(ptr *Ctensor, sizeData []int64, sizeLen int, fillValue Cscalar, optionsKind int32, optionsDevice int32)

func AtgFullLike ¶

func AtgFullLike(ptr *Ctensor, self Ctensor, fillValue Cscalar)

func AtgFullLikeOut ¶ added in v0.8.0

func AtgFullLikeOut(ptr *Ctensor, out Ctensor, self Ctensor, fillValue Cscalar)

func AtgFullOut ¶

func AtgFullOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, fillValue Cscalar)

func AtgFusedMovingAvgObsFakeQuant ¶ added in v0.5.0

func AtgFusedMovingAvgObsFakeQuant(ptr *Ctensor, self Ctensor, observerOn Ctensor, fakeQuantOn Ctensor, runningMin Ctensor, runningMax Ctensor, scale Ctensor, zeroPoint Ctensor, averagingConst float64, quantMin int64, quantMax int64, chAxis int64, perRowFakeQuant int32, symmetricQuant int32)

func AtgGather ¶

func AtgGather(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, sparseGrad int32)

func AtgGatherBackward ¶ added in v0.3.0

func AtgGatherBackward(ptr *Ctensor, grad Ctensor, self Ctensor, dim int64, index Ctensor, sparseGrad int32)

func AtgGatherOut ¶

func AtgGatherOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, sparseGrad int32)

func AtgGcd ¶ added in v0.3.0

func AtgGcd(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGcdOut ¶ added in v0.3.0

func AtgGcdOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgGcd_ ¶ added in v0.3.0

func AtgGcd_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGe ¶

func AtgGe(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgGeScalarOut ¶ added in v0.4.0

func AtgGeScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgGeTensor ¶ added in v0.4.0

func AtgGeTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGeTensorOut ¶ added in v0.4.0

func AtgGeTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgGeTensor_ ¶ added in v0.4.0

func AtgGeTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGe_ ¶

func AtgGe_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgGelu ¶

func AtgGelu(ptr *Ctensor, self Ctensor, approximate string)

func AtgGeluBackward ¶

func AtgGeluBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, approximate string)

func AtgGeluBackwardGradInput ¶ added in v0.5.0

func AtgGeluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, approximate string)

func AtgGeluOut ¶ added in v0.5.0

func AtgGeluOut(ptr *Ctensor, out Ctensor, self Ctensor, approximate string)

func AtgGelu_ ¶ added in v0.8.0

func AtgGelu_(ptr *Ctensor, self Ctensor, approximate string)

func AtgGeometric ¶ added in v0.8.0

func AtgGeometric(ptr *Ctensor, self Ctensor, p float64)

func AtgGeometricOut ¶ added in v0.8.0

func AtgGeometricOut(ptr *Ctensor, out Ctensor, self Ctensor, p float64)

func AtgGeometric_ ¶

func AtgGeometric_(ptr *Ctensor, self Ctensor, p float64)

func AtgGeqrf ¶

func AtgGeqrf(ptr *Ctensor, self Ctensor)

func AtgGeqrfA ¶ added in v0.4.0

func AtgGeqrfA(ptr *Ctensor, a Ctensor, tau Ctensor, self Ctensor)

func AtgGer ¶

func AtgGer(ptr *Ctensor, self Ctensor, vec2 Ctensor)

func AtgGerOut ¶

func AtgGerOut(ptr *Ctensor, out Ctensor, self Ctensor, vec2 Ctensor)

func AtgGlu ¶

func AtgGlu(ptr *Ctensor, self Ctensor, dim int64)

func AtgGluBackward ¶

func AtgGluBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, dim int64)

func AtgGluBackwardGradInput ¶ added in v0.4.0

func AtgGluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, dim int64)

func AtgGluBackwardJvp ¶ added in v0.8.0

func AtgGluBackwardJvp(ptr *Ctensor, gradX Ctensor, gradGlu Ctensor, x Ctensor, dgradGlu Ctensor, dx Ctensor, dim int64)

func AtgGluBackwardJvpOut ¶ added in v0.8.0

func AtgGluBackwardJvpOut(ptr *Ctensor, out Ctensor, gradX Ctensor, gradGlu Ctensor, x Ctensor, dgradGlu Ctensor, dx Ctensor, dim int64)

func AtgGluJvp ¶ added in v0.8.0

func AtgGluJvp(ptr *Ctensor, glu Ctensor, x Ctensor, dx Ctensor, dim int64)

func AtgGluJvpOut ¶ added in v0.8.0

func AtgGluJvpOut(ptr *Ctensor, out Ctensor, glu Ctensor, x Ctensor, dx Ctensor, dim int64)

func AtgGluOut ¶

func AtgGluOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)

func AtgGrad ¶

func AtgGrad(ptr *Ctensor, self Ctensor)

func AtgGreater ¶ added in v0.3.0

func AtgGreater(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgGreaterEqual ¶ added in v0.3.0

func AtgGreaterEqual(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgGreaterEqualScalarOut ¶ added in v0.4.0

func AtgGreaterEqualScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgGreaterEqualTensor ¶ added in v0.4.0

func AtgGreaterEqualTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGreaterEqualTensorOut ¶ added in v0.4.0

func AtgGreaterEqualTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgGreaterEqualTensor_ ¶ added in v0.4.0

func AtgGreaterEqualTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGreaterEqual_ ¶ added in v0.3.0

func AtgGreaterEqual_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgGreaterScalarOut ¶ added in v0.4.0

func AtgGreaterScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgGreaterTensor ¶ added in v0.4.0

func AtgGreaterTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGreaterTensorOut ¶ added in v0.4.0

func AtgGreaterTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgGreaterTensor_ ¶ added in v0.4.0

func AtgGreaterTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGreater_ ¶ added in v0.3.0

func AtgGreater_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgGridSampler ¶

func AtgGridSampler(ptr *Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, paddingMode int64, alignCorners int32)

func AtgGridSampler2d ¶

func AtgGridSampler2d(ptr *Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, paddingMode int64, alignCorners int32)

func AtgGridSampler2dOut ¶ added in v0.8.0

func AtgGridSampler2dOut(ptr *Ctensor, out Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, paddingMode int64, alignCorners int32)

func AtgGridSampler3d ¶

func AtgGridSampler3d(ptr *Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, paddingMode int64, alignCorners int32)

func AtgGridSampler3dOut ¶ added in v0.8.0

func AtgGridSampler3dOut(ptr *Ctensor, out Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, paddingMode int64, alignCorners int32)

func AtgGroupNorm ¶

func AtgGroupNorm(ptr *Ctensor, input Ctensor, numGroups int64, weight Ctensor, bias Ctensor, eps float64, cudnnEnabled int32)

func AtgGru ¶

func AtgGru(ptr *Ctensor, input Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32, batchFirst int32)

func AtgGruCell ¶

func AtgGruCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, bHh Ctensor)

func AtgGruData ¶ added in v0.4.0

func AtgGruData(ptr *Ctensor, data Ctensor, batchSizes Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32)

func AtgGt ¶

func AtgGt(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgGtScalarOut ¶ added in v0.4.0

func AtgGtScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgGtTensor ¶ added in v0.4.0

func AtgGtTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGtTensorOut ¶ added in v0.4.0

func AtgGtTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgGtTensor_ ¶ added in v0.4.0

func AtgGtTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgGt_ ¶

func AtgGt_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgHammingWindow ¶

func AtgHammingWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)

func AtgHammingWindowOut ¶ added in v0.8.0

func AtgHammingWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)

func AtgHammingWindowPeriodic ¶ added in v0.4.0

func AtgHammingWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, optionsDevice int32)

func AtgHammingWindowPeriodicAlpha ¶ added in v0.4.0

func AtgHammingWindowPeriodicAlpha(ptr *Ctensor, windowLength int64, periodic int32, alpha float64, optionsKind int32, optionsDevice int32)

func AtgHammingWindowPeriodicAlphaBeta ¶ added in v0.4.0

func AtgHammingWindowPeriodicAlphaBeta(ptr *Ctensor, windowLength int64, periodic int32, alpha float64, beta float64, optionsKind int32, optionsDevice int32)

func AtgHammingWindowPeriodicAlphaBetaOut ¶ added in v0.8.0

func AtgHammingWindowPeriodicAlphaBetaOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32, alpha float64, beta float64)

func AtgHammingWindowPeriodicAlphaOut ¶ added in v0.8.0

func AtgHammingWindowPeriodicAlphaOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32, alpha float64)

func AtgHammingWindowPeriodicOut ¶ added in v0.8.0

func AtgHammingWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)

func AtgHannWindow ¶

func AtgHannWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)

func AtgHannWindowOut ¶ added in v0.8.0

func AtgHannWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)

func AtgHannWindowPeriodic ¶ added in v0.4.0

func AtgHannWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, optionsDevice int32)

func AtgHannWindowPeriodicOut ¶ added in v0.8.0

func AtgHannWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)

func AtgHardshrink ¶

func AtgHardshrink(ptr *Ctensor, self Ctensor)

func AtgHardshrinkBackward ¶

func AtgHardshrinkBackward(ptr *Ctensor, gradOut Ctensor, self Ctensor, lambd Cscalar)

func AtgHardshrinkBackwardGradInput ¶ added in v0.5.0

func AtgHardshrinkBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOut Ctensor, self Ctensor, lambd Cscalar)

func AtgHardshrinkOut ¶ added in v0.5.0

func AtgHardshrinkOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgHardsigmoid ¶

func AtgHardsigmoid(ptr *Ctensor, self Ctensor)

func AtgHardsigmoidBackward ¶

func AtgHardsigmoidBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)

func AtgHardsigmoidBackwardGradInput ¶ added in v0.5.0

func AtgHardsigmoidBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor)

func AtgHardsigmoidOut ¶

func AtgHardsigmoidOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgHardsigmoid_ ¶

func AtgHardsigmoid_(ptr *Ctensor, self Ctensor)

func AtgHardswish ¶ added in v0.3.0

func AtgHardswish(ptr *Ctensor, self Ctensor)

func AtgHardswishBackward ¶ added in v0.3.0

func AtgHardswishBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)

func AtgHardswishBackwardOut ¶ added in v0.8.0

func AtgHardswishBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor)

func AtgHardswishOut ¶ added in v0.3.0

func AtgHardswishOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgHardswish_ ¶ added in v0.3.0

func AtgHardswish_(ptr *Ctensor, self Ctensor)

func AtgHardtanh ¶

func AtgHardtanh(ptr *Ctensor, self Ctensor)

func AtgHardtanhBackward ¶

func AtgHardtanhBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, minVal Cscalar, maxVal Cscalar)

func AtgHardtanhBackwardGradInput ¶ added in v0.4.0

func AtgHardtanhBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, minVal Cscalar, maxVal Cscalar)

func AtgHardtanhOut ¶

func AtgHardtanhOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgHardtanh_ ¶

func AtgHardtanh_(ptr *Ctensor, self Ctensor)

func AtgHeaviside ¶ added in v0.3.0

func AtgHeaviside(ptr *Ctensor, self Ctensor, values Ctensor)

func AtgHeavisideOut ¶ added in v0.3.0

func AtgHeavisideOut(ptr *Ctensor, out Ctensor, self Ctensor, values Ctensor)

func AtgHeaviside_ ¶ added in v0.3.0

func AtgHeaviside_(ptr *Ctensor, self Ctensor, values Ctensor)

func AtgHingeEmbeddingLoss ¶

func AtgHingeEmbeddingLoss(ptr *Ctensor, self Ctensor, target Ctensor, margin float64, reduction int64)

func AtgHistc ¶

func AtgHistc(ptr *Ctensor, self Ctensor, bins int64)

func AtgHistcOut ¶

func AtgHistcOut(ptr *Ctensor, out Ctensor, self Ctensor, bins int64)

func AtgHspmm ¶

func AtgHspmm(ptr *Ctensor, mat1 Ctensor, mat2 Ctensor)

func AtgHspmmOut ¶

func AtgHspmmOut(ptr *Ctensor, out Ctensor, mat1 Ctensor, mat2 Ctensor)

func AtgHstack ¶ added in v0.3.0

func AtgHstack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgHstackOut ¶ added in v0.3.0

func AtgHstackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgHuberLoss ¶ added in v0.4.0

func AtgHuberLoss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64, delta float64)

func AtgHuberLossBackward ¶ added in v0.4.0

func AtgHuberLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64, delta float64)

func AtgHuberLossBackwardOut ¶ added in v0.4.0

func AtgHuberLossBackwardOut(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64, delta float64)

func AtgHuberLossOut ¶ added in v0.4.0

func AtgHuberLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64, delta float64)

func AtgHypot ¶ added in v0.3.0

func AtgHypot(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgHypotOut ¶ added in v0.3.0

func AtgHypotOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgHypot_ ¶ added in v0.3.0

func AtgHypot_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgI0 ¶ added in v0.3.0

func AtgI0(ptr *Ctensor, self Ctensor)

func AtgI0Out ¶ added in v0.3.0

func AtgI0Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgI0_ ¶ added in v0.3.0

func AtgI0_(ptr *Ctensor, self Ctensor)

func AtgIgamma ¶ added in v0.4.0

func AtgIgamma(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgIgammaOut ¶ added in v0.4.0

func AtgIgammaOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgIgamma_ ¶ added in v0.4.0

func AtgIgamma_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgIgammac ¶ added in v0.4.0

func AtgIgammac(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgIgammacOut ¶ added in v0.4.0

func AtgIgammacOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgIgammac_ ¶ added in v0.4.0

func AtgIgammac_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgIm2col ¶

func AtgIm2col(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, dilationData []int64, dilationLen int, paddingData []int64, paddingLen int, strideData []int64, strideLen int)

func AtgIm2colOut ¶

func AtgIm2colOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, dilationData []int64, dilationLen int, paddingData []int64, paddingLen int, strideData []int64, strideLen int)

func AtgImag ¶

func AtgImag(ptr *Ctensor, self Ctensor)

func AtgIndex ¶

func AtgIndex(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int)

func AtgIndexAdd ¶

func AtgIndexAdd(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)

func AtgIndexAddOut ¶ added in v0.7.0

func AtgIndexAddOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)

func AtgIndexAdd_ ¶

func AtgIndexAdd_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)

func AtgIndexCopy ¶

func AtgIndexCopy(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)

func AtgIndexCopyOut ¶ added in v0.8.0

func AtgIndexCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)

func AtgIndexCopy_ ¶

func AtgIndexCopy_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor)

func AtgIndexFill ¶

func AtgIndexFill(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)

func AtgIndexFillIntScalarOut ¶ added in v0.8.0

func AtgIndexFillIntScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)

func AtgIndexFillIntTensor ¶ added in v0.4.0

func AtgIndexFillIntTensor(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Ctensor)

func AtgIndexFillIntTensorOut ¶ added in v0.8.0

func AtgIndexFillIntTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, value Ctensor)

func AtgIndexFillIntTensor_ ¶ added in v0.4.0

func AtgIndexFillIntTensor_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Ctensor)

func AtgIndexFill_ ¶

func AtgIndexFill_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)

func AtgIndexPut ¶

func AtgIndexPut(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32)

func AtgIndexPutOut ¶ added in v0.8.0

func AtgIndexPutOut(ptr *Ctensor, out Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32)

func AtgIndexPut_ ¶

func AtgIndexPut_(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32)

func AtgIndexReduce ¶ added in v0.8.0

func AtgIndexReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor, reduce string, includeSelf int32)

func AtgIndexReduceOut ¶ added in v0.8.0

func AtgIndexReduceOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor, reduce string, includeSelf int32)

func AtgIndexReduce_ ¶ added in v0.8.0

func AtgIndexReduce_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor, reduce string, includeSelf int32)

func AtgIndexSelect ¶

func AtgIndexSelect(ptr *Ctensor, self Ctensor, dim int64, index Ctensor)

func AtgIndexSelectBackward ¶ added in v0.3.0

func AtgIndexSelectBackward(ptr *Ctensor, grad Ctensor, selfSizesData []int64, selfSizesLen int, dim int64, index Ctensor)

func AtgIndexSelectOut ¶

func AtgIndexSelectOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor)

func AtgIndexTensorOut ¶ added in v0.8.0

func AtgIndexTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int)

func AtgIndices ¶

func AtgIndices(ptr *Ctensor, self Ctensor)

func AtgIndicesCopy ¶ added in v0.8.0

func AtgIndicesCopy(ptr *Ctensor, self Ctensor)

func AtgIndicesCopyOut ¶ added in v0.8.0

func AtgIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgInfinitelyDifferentiableGeluBackward ¶ added in v0.3.0

func AtgInfinitelyDifferentiableGeluBackward(ptr *Ctensor, grad Ctensor, self Ctensor)

func AtgInner ¶ added in v0.4.0

func AtgInner(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgInnerOut ¶ added in v0.4.0

func AtgInnerOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgInstanceNorm ¶

func AtgInstanceNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, useInputStats int32, momentum float64, eps float64, cudnnEnabled int32)

func AtgIntRepr ¶

func AtgIntRepr(ptr *Ctensor, self Ctensor)

func AtgIntReprOut ¶ added in v0.8.0

func AtgIntReprOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgInverse ¶

func AtgInverse(ptr *Ctensor, self Ctensor)

func AtgInverseOut ¶

func AtgInverseOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgIsCoalesced ¶ added in v0.4.0

func AtgIsCoalesced(self Ctensor) bool

func AtgIsComplex ¶ added in v0.4.0

func AtgIsComplex(self Ctensor) bool

func AtgIsConj ¶ added in v0.5.0

func AtgIsConj(self Ctensor) bool

func AtgIsDistributed ¶ added in v0.4.0

func AtgIsDistributed(self Ctensor) bool

func AtgIsFloatingPoint ¶ added in v0.4.0

func AtgIsFloatingPoint(self Ctensor) bool

func AtgIsInference ¶ added in v0.5.0

func AtgIsInference(self Ctensor) bool

func AtgIsLeaf ¶ added in v0.4.0

func AtgIsLeaf(self Ctensor) bool

func AtgIsNeg ¶ added in v0.5.0

func AtgIsNeg(self Ctensor) bool

func AtgIsNonzero ¶ added in v0.4.0

func AtgIsNonzero(self Ctensor) bool

func AtgIsPinned ¶ added in v0.4.0

func AtgIsPinned(self Ctensor, device int32) bool

func AtgIsSameSize ¶ added in v0.4.0

func AtgIsSameSize(self Ctensor, other Ctensor) bool

func AtgIsSetTo ¶ added in v0.4.0

func AtgIsSetTo(self Ctensor, tensor Ctensor) bool

func AtgIsSigned ¶ added in v0.4.0

func AtgIsSigned(self Ctensor) bool

func AtgIsVulkanAvailable ¶ added in v0.4.0

func AtgIsVulkanAvailable() bool

func AtgIsclose ¶

func AtgIsclose(ptr *Ctensor, self Ctensor, other Ctensor, rtol float64, atol float64, equalNan int32)

func AtgIsfinite ¶

func AtgIsfinite(ptr *Ctensor, self Ctensor)

func AtgIsin ¶ added in v0.5.0

func AtgIsin(ptr *Ctensor, elements Ctensor, testElements Ctensor, assumeUnique int32, invert int32)

func AtgIsinScalarTensor ¶ added in v0.5.0

func AtgIsinScalarTensor(ptr *Ctensor, element Cscalar, testElements Ctensor, assumeUnique int32, invert int32)

func AtgIsinScalarTensorOut ¶ added in v0.5.0

func AtgIsinScalarTensorOut(ptr *Ctensor, out Ctensor, element Cscalar, testElements Ctensor, assumeUnique int32, invert int32)

func AtgIsinTensorScalar ¶ added in v0.5.0

func AtgIsinTensorScalar(ptr *Ctensor, elements Ctensor, testElement Cscalar, assumeUnique int32, invert int32)

func AtgIsinTensorScalarOut ¶ added in v0.5.0

func AtgIsinTensorScalarOut(ptr *Ctensor, out Ctensor, elements Ctensor, testElement Cscalar, assumeUnique int32, invert int32)

func AtgIsinTensorTensorOut ¶ added in v0.5.0

func AtgIsinTensorTensorOut(ptr *Ctensor, out Ctensor, elements Ctensor, testElements Ctensor, assumeUnique int32, invert int32)

func AtgIsinf ¶

func AtgIsinf(ptr *Ctensor, self Ctensor)

func AtgIsinfOut ¶ added in v0.8.0

func AtgIsinfOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgIsnan ¶

func AtgIsnan(ptr *Ctensor, self Ctensor)

func AtgIsnanOut ¶ added in v0.8.0

func AtgIsnanOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgIsneginf ¶ added in v0.3.0

func AtgIsneginf(ptr *Ctensor, self Ctensor)

func AtgIsneginfOut ¶ added in v0.3.0

func AtgIsneginfOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgIsposinf ¶ added in v0.3.0

func AtgIsposinf(ptr *Ctensor, self Ctensor)

func AtgIsposinfOut ¶ added in v0.3.0

func AtgIsposinfOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgIsreal ¶ added in v0.3.0

func AtgIsreal(ptr *Ctensor, self Ctensor)

func AtgIstft ¶ added in v0.3.0

func AtgIstft(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, winLengthVal int64, winLengthNull int, window Ctensor, center int32, normalized int32, onesided int32, lengthVal int64, lengthNull int, returnComplex int32)

func AtgKaiserWindow ¶ added in v0.3.0

func AtgKaiserWindow(ptr *Ctensor, windowLength int64, optionsKind int32, optionsDevice int32)

func AtgKaiserWindowBeta ¶ added in v0.4.0

func AtgKaiserWindowBeta(ptr *Ctensor, windowLength int64, periodic int32, beta float64, optionsKind int32, optionsDevice int32)

func AtgKaiserWindowBetaOut ¶ added in v0.8.0

func AtgKaiserWindowBetaOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32, beta float64)

func AtgKaiserWindowOut ¶ added in v0.8.0

func AtgKaiserWindowOut(ptr *Ctensor, out Ctensor, windowLength int64)

func AtgKaiserWindowPeriodic ¶ added in v0.4.0

func AtgKaiserWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, optionsDevice int32)

func AtgKaiserWindowPeriodicOut ¶ added in v0.8.0

func AtgKaiserWindowPeriodicOut(ptr *Ctensor, out Ctensor, windowLength int64, periodic int32)

func AtgKlDiv ¶

func AtgKlDiv(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64, logTarget int32)

func AtgKron ¶ added in v0.4.0

func AtgKron(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgKronOut ¶ added in v0.4.0

func AtgKronOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgKthvalue ¶

func AtgKthvalue(ptr *Ctensor, self Ctensor, k int64, dim int64, keepdim int32)

func AtgKthvalueValues ¶ added in v0.4.0

func AtgKthvalueValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, k int64, dim int64, keepdim int32)

func AtgL1Loss ¶

func AtgL1Loss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgLayerNorm ¶

func AtgLayerNorm(ptr *Ctensor, input Ctensor, normalizedShapeData []int64, normalizedShapeLen int, weight Ctensor, bias Ctensor, eps float64, cudnnEnable int32)

func AtgLcm ¶ added in v0.3.0

func AtgLcm(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLcmOut ¶ added in v0.3.0

func AtgLcmOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLcm_ ¶ added in v0.3.0

func AtgLcm_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLdexp ¶ added in v0.4.0

func AtgLdexp(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLdexpOut ¶ added in v0.4.0

func AtgLdexpOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLdexp_ ¶ added in v0.4.0

func AtgLdexp_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLe ¶

func AtgLe(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgLeScalarOut ¶ added in v0.4.0

func AtgLeScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgLeTensor ¶ added in v0.4.0

func AtgLeTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLeTensorOut ¶ added in v0.4.0

func AtgLeTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLeTensor_ ¶ added in v0.4.0

func AtgLeTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLe_ ¶

func AtgLe_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgLeakyRelu ¶

func AtgLeakyRelu(ptr *Ctensor, self Ctensor)

func AtgLeakyReluBackward ¶

func AtgLeakyReluBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, negativeSlope Cscalar, selfIsResult int32)

func AtgLeakyReluBackwardGradInput ¶ added in v0.5.0

func AtgLeakyReluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, negativeSlope Cscalar, selfIsResult int32)

func AtgLeakyReluOut ¶

func AtgLeakyReluOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLeakyRelu_ ¶

func AtgLeakyRelu_(ptr *Ctensor, self Ctensor)

func AtgLerp ¶

func AtgLerp(ptr *Ctensor, self Ctensor, end Ctensor, weight Cscalar)

func AtgLerpScalarOut ¶ added in v0.4.0

func AtgLerpScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, end Ctensor, weight Cscalar)

func AtgLerpTensor ¶ added in v0.4.0

func AtgLerpTensor(ptr *Ctensor, self Ctensor, end Ctensor, weight Ctensor)

func AtgLerpTensorOut ¶ added in v0.4.0

func AtgLerpTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, end Ctensor, weight Ctensor)

func AtgLerpTensor_ ¶ added in v0.4.0

func AtgLerpTensor_(ptr *Ctensor, self Ctensor, end Ctensor, weight Ctensor)

func AtgLerp_ ¶

func AtgLerp_(ptr *Ctensor, self Ctensor, end Ctensor, weight Cscalar)

func AtgLess ¶ added in v0.3.0

func AtgLess(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgLessEqual ¶ added in v0.3.0

func AtgLessEqual(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgLessEqualScalarOut ¶ added in v0.4.0

func AtgLessEqualScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgLessEqualTensor ¶ added in v0.4.0

func AtgLessEqualTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLessEqualTensorOut ¶ added in v0.4.0

func AtgLessEqualTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLessEqualTensor_ ¶ added in v0.4.0

func AtgLessEqualTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLessEqual_ ¶ added in v0.3.0

func AtgLessEqual_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgLessScalarOut ¶ added in v0.4.0

func AtgLessScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgLessTensor ¶ added in v0.4.0

func AtgLessTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLessTensorOut ¶ added in v0.4.0

func AtgLessTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLessTensor_ ¶ added in v0.4.0

func AtgLessTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLess_ ¶ added in v0.3.0

func AtgLess_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgLgamma ¶

func AtgLgamma(ptr *Ctensor, self Ctensor)

func AtgLgammaOut ¶

func AtgLgammaOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLgamma_ ¶

func AtgLgamma_(ptr *Ctensor, self Ctensor)

func AtgLift ¶ added in v0.8.0

func AtgLift(ptr *Ctensor, self Ctensor)

func AtgLiftFresh ¶ added in v0.8.0

func AtgLiftFresh(ptr *Ctensor, self Ctensor)

func AtgLiftFreshCopy ¶ added in v0.8.0

func AtgLiftFreshCopy(ptr *Ctensor, self Ctensor)

func AtgLiftFreshCopyOut ¶ added in v0.8.0

func AtgLiftFreshCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLiftOut ¶ added in v0.8.0

func AtgLiftOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLinalgCholesky ¶ added in v0.4.0

func AtgLinalgCholesky(ptr *Ctensor, self Ctensor, upper int32)

func AtgLinalgCholeskyEx ¶ added in v0.4.0

func AtgLinalgCholeskyEx(ptr *Ctensor, self Ctensor, upper int32, checkErrors int32)

func AtgLinalgCholeskyExL ¶ added in v0.4.0

func AtgLinalgCholeskyExL(ptr *Ctensor, l Ctensor, info Ctensor, self Ctensor, upper int32, checkErrors int32)

func AtgLinalgCholeskyOut ¶ added in v0.4.0

func AtgLinalgCholeskyOut(ptr *Ctensor, out Ctensor, self Ctensor, upper int32)

func AtgLinalgCond ¶ added in v0.4.0

func AtgLinalgCond(ptr *Ctensor, self Ctensor, p Cscalar)

func AtgLinalgCondOut ¶ added in v0.4.0

func AtgLinalgCondOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar)

func AtgLinalgCondPStr ¶ added in v0.4.0

func AtgLinalgCondPStr(ptr *Ctensor, self Ctensor, p string)

func AtgLinalgCondPStrOut ¶ added in v0.4.0

func AtgLinalgCondPStrOut(ptr *Ctensor, out Ctensor, self Ctensor, p string)

func AtgLinalgCross ¶ added in v0.7.0

func AtgLinalgCross(ptr *Ctensor, self Ctensor, other Ctensor, dim int64)

func AtgLinalgCrossOut ¶ added in v0.7.0

func AtgLinalgCrossOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, dim int64)

func AtgLinalgDet ¶ added in v0.3.0

func AtgLinalgDet(ptr *Ctensor, a Ctensor)

func AtgLinalgDetOut ¶ added in v0.4.0

func AtgLinalgDetOut(ptr *Ctensor, out Ctensor, a Ctensor)

func AtgLinalgDiagonal ¶ added in v0.7.0

func AtgLinalgDiagonal(ptr *Ctensor, a Ctensor, offset int64, dim1 int64, dim2 int64)

func AtgLinalgEig ¶ added in v0.4.0

func AtgLinalgEig(ptr *Ctensor, self Ctensor)

func AtgLinalgEigOut ¶ added in v0.4.0

func AtgLinalgEigOut(ptr *Ctensor, eigenvalues Ctensor, eigenvectors Ctensor, self Ctensor)

func AtgLinalgEigh ¶ added in v0.4.0

func AtgLinalgEigh(ptr *Ctensor, self Ctensor, uPLO string)

func AtgLinalgEighEigvals ¶ added in v0.4.0

func AtgLinalgEighEigvals(ptr *Ctensor, eigvals Ctensor, eigvecs Ctensor, self Ctensor, uPLO string)

func AtgLinalgEigvals ¶ added in v0.4.0

func AtgLinalgEigvals(ptr *Ctensor, self Ctensor)

func AtgLinalgEigvalsOut ¶ added in v0.4.0

func AtgLinalgEigvalsOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLinalgEigvalsh ¶ added in v0.4.0

func AtgLinalgEigvalsh(ptr *Ctensor, self Ctensor, uPLO string)

func AtgLinalgEigvalshOut ¶ added in v0.4.0

func AtgLinalgEigvalshOut(ptr *Ctensor, out Ctensor, self Ctensor, uPLO string)

func AtgLinalgHouseholderProduct ¶ added in v0.4.0

func AtgLinalgHouseholderProduct(ptr *Ctensor, input Ctensor, tau Ctensor)

func AtgLinalgHouseholderProductOut ¶ added in v0.4.0

func AtgLinalgHouseholderProductOut(ptr *Ctensor, out Ctensor, input Ctensor, tau Ctensor)

func AtgLinalgInv ¶ added in v0.4.0

func AtgLinalgInv(ptr *Ctensor, a Ctensor)

func AtgLinalgInvEx ¶ added in v0.4.0

func AtgLinalgInvEx(ptr *Ctensor, a Ctensor, checkErrors int32)

func AtgLinalgInvExInverse ¶ added in v0.4.0

func AtgLinalgInvExInverse(ptr *Ctensor, inverse Ctensor, info Ctensor, a Ctensor, checkErrors int32)

func AtgLinalgInvOut ¶ added in v0.4.0

func AtgLinalgInvOut(ptr *Ctensor, out Ctensor, a Ctensor)

func AtgLinalgLdlFactor ¶ added in v0.8.0

func AtgLinalgLdlFactor(ptr *Ctensor, self Ctensor, hermitian int32)

func AtgLinalgLdlFactorEx ¶ added in v0.8.0

func AtgLinalgLdlFactorEx(ptr *Ctensor, self Ctensor, hermitian int32, checkErrors int32)

func AtgLinalgLdlFactorExOut ¶ added in v0.8.0

func AtgLinalgLdlFactorExOut(ptr *Ctensor, lD Ctensor, pivots Ctensor, info Ctensor, self Ctensor, hermitian int32, checkErrors int32)

func AtgLinalgLdlFactorOut ¶ added in v0.8.0

func AtgLinalgLdlFactorOut(ptr *Ctensor, lD Ctensor, pivots Ctensor, self Ctensor, hermitian int32)

func AtgLinalgLdlSolve ¶ added in v0.8.0

func AtgLinalgLdlSolve(ptr *Ctensor, lD Ctensor, pivots Ctensor, b Ctensor, hermitian int32)

func AtgLinalgLdlSolveOut ¶ added in v0.8.0

func AtgLinalgLdlSolveOut(ptr *Ctensor, out Ctensor, lD Ctensor, pivots Ctensor, b Ctensor, hermitian int32)

func AtgLinalgLstsq ¶ added in v0.4.0

func AtgLinalgLstsq(ptr *Ctensor, self Ctensor, b Ctensor, rcondVal float64, rcondNull int, driver string)

func AtgLinalgLstsqOut ¶ added in v0.4.0

func AtgLinalgLstsqOut(ptr *Ctensor, solution Ctensor, residuals Ctensor, rank Ctensor, singularValues Ctensor, self Ctensor, b Ctensor, rcondVal float64, rcondNull int, driver string)

func AtgLinalgLu ¶ added in v0.8.0

func AtgLinalgLu(ptr *Ctensor, a Ctensor, pivot int32)

func AtgLinalgLuFactor ¶ added in v0.7.0

func AtgLinalgLuFactor(ptr *Ctensor, a Ctensor, pivot int32)

func AtgLinalgLuFactorEx ¶ added in v0.7.0

func AtgLinalgLuFactorEx(ptr *Ctensor, a Ctensor, pivot int32, checkErrors int32)

func AtgLinalgLuFactorExOut ¶ added in v0.7.0

func AtgLinalgLuFactorExOut(ptr *Ctensor, lU Ctensor, pivots Ctensor, info Ctensor, a Ctensor, pivot int32, checkErrors int32)

func AtgLinalgLuFactorOut ¶ added in v0.7.0

func AtgLinalgLuFactorOut(ptr *Ctensor, lU Ctensor, pivots Ctensor, a Ctensor, pivot int32)

func AtgLinalgLuOut ¶ added in v0.8.0

func AtgLinalgLuOut(ptr *Ctensor, p Ctensor, l Ctensor, u Ctensor, a Ctensor, pivot int32)

func AtgLinalgLuSolve ¶ added in v0.8.0

func AtgLinalgLuSolve(ptr *Ctensor, lU Ctensor, pivots Ctensor, b Ctensor, left int32, adjoint int32)

func AtgLinalgLuSolveOut ¶ added in v0.8.0

func AtgLinalgLuSolveOut(ptr *Ctensor, out Ctensor, lU Ctensor, pivots Ctensor, b Ctensor, left int32, adjoint int32)

func AtgLinalgMatmul ¶ added in v0.5.0

func AtgLinalgMatmul(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLinalgMatmulOut ¶ added in v0.5.0

func AtgLinalgMatmulOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLinalgMatrixExp ¶ added in v0.7.0

func AtgLinalgMatrixExp(ptr *Ctensor, self Ctensor)

func AtgLinalgMatrixExpOut ¶ added in v0.8.0

func AtgLinalgMatrixExpOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLinalgMatrixPower ¶ added in v0.4.0

func AtgLinalgMatrixPower(ptr *Ctensor, self Ctensor, n int64)

func AtgLinalgMatrixPowerOut ¶ added in v0.4.0

func AtgLinalgMatrixPowerOut(ptr *Ctensor, out Ctensor, self Ctensor, n int64)

func AtgLinalgMatrixRank ¶ added in v0.4.0

func AtgLinalgMatrixRank(ptr *Ctensor, self Ctensor, tol float64, hermitian int32)

func AtgLinalgMatrixRankAtolRtolFloat ¶ added in v0.7.0

func AtgLinalgMatrixRankAtolRtolFloat(ptr *Ctensor, self Ctensor, atolVal float64, atolNull int, rtolVal float64, rtolNull int, hermitian int32)

func AtgLinalgMatrixRankAtolRtolFloatOut ¶ added in v0.7.0

func AtgLinalgMatrixRankAtolRtolFloatOut(ptr *Ctensor, out Ctensor, self Ctensor, atolVal float64, atolNull int, rtolVal float64, rtolNull int, hermitian int32)

func AtgLinalgMatrixRankAtolRtolTensor ¶ added in v0.7.0

func AtgLinalgMatrixRankAtolRtolTensor(ptr *Ctensor, input Ctensor, atol Ctensor, rtol Ctensor, hermitian int32)

func AtgLinalgMatrixRankAtolRtolTensorOut ¶ added in v0.7.0

func AtgLinalgMatrixRankAtolRtolTensorOut(ptr *Ctensor, out Ctensor, input Ctensor, atol Ctensor, rtol Ctensor, hermitian int32)

func AtgLinalgMatrixRankOut ¶ added in v0.4.0

func AtgLinalgMatrixRankOut(ptr *Ctensor, out Ctensor, self Ctensor, tol float64, hermitian int32)

func AtgLinalgMatrixRankOutTolTensor ¶ added in v0.4.0

func AtgLinalgMatrixRankOutTolTensor(ptr *Ctensor, out Ctensor, input Ctensor, tol Ctensor, hermitian int32)

func AtgLinalgMatrixRankTolTensor ¶ added in v0.4.0

func AtgLinalgMatrixRankTolTensor(ptr *Ctensor, input Ctensor, tol Ctensor, hermitian int32)

func AtgLinalgMultiDot ¶ added in v0.4.0

func AtgLinalgMultiDot(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgLinalgMultiDotOut ¶ added in v0.4.0

func AtgLinalgMultiDotOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgLinalgNorm ¶ added in v0.3.0

func AtgLinalgNorm(ptr *Ctensor, self Ctensor, ord Cscalar, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgLinalgNormOrdStr ¶ added in v0.4.0

func AtgLinalgNormOrdStr(ptr *Ctensor, self Ctensor, ord string, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgLinalgNormOrdStrOut ¶ added in v0.4.0

func AtgLinalgNormOrdStrOut(ptr *Ctensor, out Ctensor, self Ctensor, ord string, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgLinalgNormOut ¶ added in v0.3.0

func AtgLinalgNormOut(ptr *Ctensor, out Ctensor, self Ctensor, ord Cscalar, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgLinalgPinv ¶ added in v0.4.0

func AtgLinalgPinv(ptr *Ctensor, self Ctensor, rcond float64, hermitian int32)

func AtgLinalgPinvAtolRtolFloat ¶ added in v0.7.0

func AtgLinalgPinvAtolRtolFloat(ptr *Ctensor, self Ctensor, atolVal float64, atolNull int, rtolVal float64, rtolNull int, hermitian int32)

func AtgLinalgPinvAtolRtolFloatOut ¶ added in v0.7.0

func AtgLinalgPinvAtolRtolFloatOut(ptr *Ctensor, out Ctensor, self Ctensor, atolVal float64, atolNull int, rtolVal float64, rtolNull int, hermitian int32)

func AtgLinalgPinvAtolRtolTensor ¶ added in v0.7.0

func AtgLinalgPinvAtolRtolTensor(ptr *Ctensor, self Ctensor, atol Ctensor, rtol Ctensor, hermitian int32)

func AtgLinalgPinvAtolRtolTensorOut ¶ added in v0.7.0

func AtgLinalgPinvAtolRtolTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, atol Ctensor, rtol Ctensor, hermitian int32)

func AtgLinalgPinvOut ¶ added in v0.4.0

func AtgLinalgPinvOut(ptr *Ctensor, out Ctensor, self Ctensor, rcond float64, hermitian int32)

func AtgLinalgPinvOutRcondTensor ¶ added in v0.4.0

func AtgLinalgPinvOutRcondTensor(ptr *Ctensor, out Ctensor, self Ctensor, rcond Ctensor, hermitian int32)

func AtgLinalgPinvRcondTensor ¶ added in v0.4.0

func AtgLinalgPinvRcondTensor(ptr *Ctensor, self Ctensor, rcond Ctensor, hermitian int32)

func AtgLinalgQr ¶ added in v0.4.0

func AtgLinalgQr(ptr *Ctensor, a Ctensor, mode string)

func AtgLinalgQrOut ¶ added in v0.4.0

func AtgLinalgQrOut(ptr *Ctensor, q Ctensor, r Ctensor, a Ctensor, mode string)

func AtgLinalgSlogdet ¶ added in v0.4.0

func AtgLinalgSlogdet(ptr *Ctensor, a Ctensor)

func AtgLinalgSlogdetOut ¶ added in v0.4.0

func AtgLinalgSlogdetOut(ptr *Ctensor, sign Ctensor, logabsdet Ctensor, a Ctensor)

func AtgLinalgSolve ¶ added in v0.4.0

func AtgLinalgSolve(ptr *Ctensor, a Ctensor, b Ctensor, left int32)

func AtgLinalgSolveEx ¶ added in v0.8.0

func AtgLinalgSolveEx(ptr *Ctensor, a Ctensor, b Ctensor, left int32, checkErrors int32)

func AtgLinalgSolveExOut ¶ added in v0.8.0

func AtgLinalgSolveExOut(ptr *Ctensor, result Ctensor, info Ctensor, a Ctensor, b Ctensor, left int32, checkErrors int32)

func AtgLinalgSolveOut ¶ added in v0.4.0

func AtgLinalgSolveOut(ptr *Ctensor, out Ctensor, a Ctensor, b Ctensor, left int32)

func AtgLinalgSolveTriangular ¶ added in v0.7.0

func AtgLinalgSolveTriangular(ptr *Ctensor, self Ctensor, b Ctensor, upper int32, left int32, unitriangular int32)

func AtgLinalgSolveTriangularOut ¶ added in v0.7.0

func AtgLinalgSolveTriangularOut(ptr *Ctensor, out Ctensor, self Ctensor, b Ctensor, upper int32, left int32, unitriangular int32)

func AtgLinalgSvd ¶ added in v0.4.0

func AtgLinalgSvd(ptr *Ctensor, a Ctensor, fullMatrices int32, driver string)

func AtgLinalgSvdU ¶ added in v0.4.0

func AtgLinalgSvdU(ptr *Ctensor, u Ctensor, s Ctensor, vh Ctensor, a Ctensor, fullMatrices int32, driver string)

func AtgLinalgSvdvals ¶ added in v0.4.0

func AtgLinalgSvdvals(ptr *Ctensor, a Ctensor, driver string)

func AtgLinalgSvdvalsOut ¶ added in v0.4.0

func AtgLinalgSvdvalsOut(ptr *Ctensor, out Ctensor, a Ctensor, driver string)

func AtgLinalgTensorinv ¶ added in v0.4.0

func AtgLinalgTensorinv(ptr *Ctensor, self Ctensor, ind int64)

func AtgLinalgTensorinvOut ¶ added in v0.4.0

func AtgLinalgTensorinvOut(ptr *Ctensor, out Ctensor, self Ctensor, ind int64)

func AtgLinalgTensorsolve ¶ added in v0.4.0

func AtgLinalgTensorsolve(ptr *Ctensor, self Ctensor, other Ctensor, dimsData []int64, dimsLen int)

func AtgLinalgTensorsolveOut ¶ added in v0.4.0

func AtgLinalgTensorsolveOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, dimsData []int64, dimsLen int)

func AtgLinalgVander ¶ added in v0.8.0

func AtgLinalgVander(ptr *Ctensor, x Ctensor, nVal int64, nNull int)

func AtgLinalgVecdot ¶ added in v0.8.0

func AtgLinalgVecdot(ptr *Ctensor, x Ctensor, y Ctensor, dim int64)

func AtgLinalgVecdotOut ¶ added in v0.8.0

func AtgLinalgVecdotOut(ptr *Ctensor, out Ctensor, x Ctensor, y Ctensor, dim int64)

func AtgLinear ¶

func AtgLinear(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor)

func AtgLinearOut ¶ added in v0.5.0

func AtgLinearOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor)

func AtgLinspace ¶

func AtgLinspace(ptr *Ctensor, start Cscalar, end Cscalar, steps int64, optionsKind int32, optionsDevice int32)

func AtgLinspaceOut ¶

func AtgLinspaceOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar, steps int64)

func AtgLog ¶

func AtgLog(ptr *Ctensor, self Ctensor)

func AtgLog10 ¶

func AtgLog10(ptr *Ctensor, self Ctensor)

func AtgLog10Out ¶

func AtgLog10Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLog10_ ¶

func AtgLog10_(ptr *Ctensor, self Ctensor)

func AtgLog1p ¶

func AtgLog1p(ptr *Ctensor, self Ctensor)

func AtgLog1pOut ¶

func AtgLog1pOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLog1p_ ¶

func AtgLog1p_(ptr *Ctensor, self Ctensor)

func AtgLog2 ¶

func AtgLog2(ptr *Ctensor, self Ctensor)

func AtgLog2Out ¶

func AtgLog2Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLog2_ ¶

func AtgLog2_(ptr *Ctensor, self Ctensor)

func AtgLogNormal ¶ added in v0.8.0

func AtgLogNormal(ptr *Ctensor, self Ctensor, mean float64, std float64)

func AtgLogNormalOut ¶ added in v0.8.0

func AtgLogNormalOut(ptr *Ctensor, out Ctensor, self Ctensor, mean float64, std float64)

func AtgLogNormal_ ¶

func AtgLogNormal_(ptr *Ctensor, self Ctensor, mean float64, std float64)

func AtgLogOut ¶

func AtgLogOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLogSigmoid ¶

func AtgLogSigmoid(ptr *Ctensor, self Ctensor)

func AtgLogSigmoidBackward ¶

func AtgLogSigmoidBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, buffer Ctensor)

func AtgLogSigmoidBackwardGradInput ¶ added in v0.4.0

func AtgLogSigmoidBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, buffer Ctensor)

func AtgLogSigmoidOut ¶

func AtgLogSigmoidOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLogSoftmax ¶

func AtgLogSoftmax(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func AtgLogSoftmaxIntOut ¶ added in v0.8.0

func AtgLogSoftmaxIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, dtype int32)

func AtgLog_ ¶

func AtgLog_(ptr *Ctensor, self Ctensor)

func AtgLogaddexp ¶ added in v0.3.0

func AtgLogaddexp(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLogaddexp2 ¶ added in v0.3.0

func AtgLogaddexp2(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLogaddexp2Out ¶ added in v0.3.0

func AtgLogaddexp2Out(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLogaddexpOut ¶ added in v0.3.0

func AtgLogaddexpOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLogcumsumexp ¶ added in v0.3.0

func AtgLogcumsumexp(ptr *Ctensor, self Ctensor, dim int64)

func AtgLogcumsumexpOut ¶ added in v0.3.0

func AtgLogcumsumexpOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)

func AtgLogdet ¶

func AtgLogdet(ptr *Ctensor, self Ctensor)

func AtgLogicalAnd ¶

func AtgLogicalAnd(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLogicalAndOut ¶

func AtgLogicalAndOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLogicalAnd_ ¶

func AtgLogicalAnd_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLogicalNot ¶

func AtgLogicalNot(ptr *Ctensor, self Ctensor)

func AtgLogicalNotOut ¶

func AtgLogicalNotOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgLogicalNot_ ¶

func AtgLogicalNot_(ptr *Ctensor, self Ctensor)

func AtgLogicalOr ¶

func AtgLogicalOr(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLogicalOrOut ¶

func AtgLogicalOrOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLogicalOr_ ¶

func AtgLogicalOr_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLogicalXor ¶

func AtgLogicalXor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLogicalXorOut ¶

func AtgLogicalXorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLogicalXor_ ¶

func AtgLogicalXor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLogit ¶ added in v0.3.0

func AtgLogit(ptr *Ctensor, self Ctensor, epsVal float64, epsNull int)

func AtgLogitBackward ¶ added in v0.3.0

func AtgLogitBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, epsVal float64, epsNull int)

func AtgLogitBackwardGradInput ¶ added in v0.4.0

func AtgLogitBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, epsVal float64, epsNull int)

func AtgLogitOut ¶ added in v0.3.0

func AtgLogitOut(ptr *Ctensor, out Ctensor, self Ctensor, epsVal float64, epsNull int)

func AtgLogit_ ¶ added in v0.3.0

func AtgLogit_(ptr *Ctensor, self Ctensor, epsVal float64, epsNull int)

func AtgLogspace ¶

func AtgLogspace(ptr *Ctensor, start Cscalar, end Cscalar, steps int64, base float64, optionsKind int32, optionsDevice int32)

func AtgLogspaceOut ¶

func AtgLogspaceOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar, steps int64, base float64)

func AtgLogsumexp ¶

func AtgLogsumexp(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgLogsumexpOut ¶

func AtgLogsumexpOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgLstm ¶

func AtgLstm(ptr *Ctensor, input Ctensor, hxData []Ctensor, hxLen int, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32, batchFirst int32)

func AtgLstmCell ¶

func AtgLstmCell(ptr *Ctensor, input Ctensor, hxData []Ctensor, hxLen int, wIh Ctensor, wHh Ctensor, bIh Ctensor, bHh Ctensor)

func AtgLstmData ¶ added in v0.4.0

func AtgLstmData(ptr *Ctensor, data Ctensor, batchSizes Ctensor, hxData []Ctensor, hxLen int, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32)

func AtgLt ¶

func AtgLt(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgLtScalarOut ¶ added in v0.4.0

func AtgLtScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgLtTensor ¶ added in v0.4.0

func AtgLtTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLtTensorOut ¶ added in v0.4.0

func AtgLtTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgLtTensor_ ¶ added in v0.4.0

func AtgLtTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgLt_ ¶

func AtgLt_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgLuSolve ¶

func AtgLuSolve(ptr *Ctensor, self Ctensor, lUData Ctensor, lUPivots Ctensor)

func AtgLuSolveOut ¶

func AtgLuSolveOut(ptr *Ctensor, out Ctensor, self Ctensor, lUData Ctensor, lUPivots Ctensor)

func AtgLuUnpack ¶ added in v0.4.0

func AtgLuUnpack(ptr *Ctensor, lUData Ctensor, lUPivots Ctensor, unpackData int32, unpackPivots int32)

func AtgLuUnpackOut ¶ added in v0.4.0

func AtgLuUnpackOut(ptr *Ctensor, p Ctensor, l Ctensor, u Ctensor, lUData Ctensor, lUPivots Ctensor, unpackData int32, unpackPivots int32)

func AtgMarginRankingLoss ¶

func AtgMarginRankingLoss(ptr *Ctensor, input1 Ctensor, input2 Ctensor, target Ctensor, margin float64, reduction int64)

func AtgMaskedFill ¶

func AtgMaskedFill(ptr *Ctensor, self Ctensor, mask Ctensor, value Cscalar)

func AtgMaskedFillScalarOut ¶ added in v0.8.0

func AtgMaskedFillScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, value Cscalar)

func AtgMaskedFillTensor ¶ added in v0.4.0

func AtgMaskedFillTensor(ptr *Ctensor, self Ctensor, mask Ctensor, value Ctensor)

func AtgMaskedFillTensorOut ¶ added in v0.8.0

func AtgMaskedFillTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, value Ctensor)

func AtgMaskedFillTensor_ ¶ added in v0.4.0

func AtgMaskedFillTensor_(ptr *Ctensor, self Ctensor, mask Ctensor, value Ctensor)

func AtgMaskedFill_ ¶

func AtgMaskedFill_(ptr *Ctensor, self Ctensor, mask Ctensor, value Cscalar)

func AtgMaskedScatter ¶

func AtgMaskedScatter(ptr *Ctensor, self Ctensor, mask Ctensor, source Ctensor)

func AtgMaskedScatterOut ¶ added in v0.8.0

func AtgMaskedScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, source Ctensor)

func AtgMaskedScatter_ ¶

func AtgMaskedScatter_(ptr *Ctensor, self Ctensor, mask Ctensor, source Ctensor)

func AtgMaskedSelect ¶

func AtgMaskedSelect(ptr *Ctensor, self Ctensor, mask Ctensor)

func AtgMaskedSelectBackward ¶ added in v0.3.0

func AtgMaskedSelectBackward(ptr *Ctensor, grad Ctensor, input Ctensor, mask Ctensor)

func AtgMaskedSelectOut ¶

func AtgMaskedSelectOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor)

func AtgMatmul ¶

func AtgMatmul(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgMatmulOut ¶

func AtgMatmulOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgMatrixExp ¶ added in v0.3.0

func AtgMatrixExp(ptr *Ctensor, self Ctensor)

func AtgMatrixExpBackward ¶ added in v0.3.0

func AtgMatrixExpBackward(ptr *Ctensor, self Ctensor, grad Ctensor)

func AtgMatrixH ¶ added in v0.7.0

func AtgMatrixH(ptr *Ctensor, self Ctensor)

func AtgMatrixPower ¶

func AtgMatrixPower(ptr *Ctensor, self Ctensor, n int64)

func AtgMatrixPowerOut ¶ added in v0.4.0

func AtgMatrixPowerOut(ptr *Ctensor, out Ctensor, self Ctensor, n int64)

func AtgMax ¶

func AtgMax(ptr *Ctensor, self Ctensor)

func AtgMaxDim ¶ added in v0.4.0

func AtgMaxDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgMaxDimMax ¶ added in v0.4.0

func AtgMaxDimMax(ptr *Ctensor, max Ctensor, maxValues Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgMaxOther ¶ added in v0.4.0

func AtgMaxOther(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgMaxOut ¶

func AtgMaxOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgMaxPool1d ¶

func AtgMaxPool1d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxPool1dWithIndices ¶

func AtgMaxPool1dWithIndices(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxPool2d ¶

func AtgMaxPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxPool2dBackward ¶ added in v0.8.0

func AtgMaxPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxPool2dBackwardOut ¶ added in v0.8.0

func AtgMaxPool2dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxPool2dWithIndices ¶

func AtgMaxPool2dWithIndices(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxPool2dWithIndicesBackward ¶

func AtgMaxPool2dWithIndicesBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32, indices Ctensor)

func AtgMaxPool2dWithIndicesBackwardGradInput ¶ added in v0.4.0

func AtgMaxPool2dWithIndicesBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32, indices Ctensor)

func AtgMaxPool2dWithIndicesOut ¶

func AtgMaxPool2dWithIndicesOut(ptr *Ctensor, out Ctensor, indices Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxPool3d ¶

func AtgMaxPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxPool3dWithIndices ¶

func AtgMaxPool3dWithIndices(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxPool3dWithIndicesBackward ¶

func AtgMaxPool3dWithIndicesBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32, indices Ctensor)

func AtgMaxPool3dWithIndicesBackwardGradInput ¶ added in v0.4.0

func AtgMaxPool3dWithIndicesBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32, indices Ctensor)

func AtgMaxPool3dWithIndicesOut ¶

func AtgMaxPool3dWithIndicesOut(ptr *Ctensor, out Ctensor, indices Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMaxUnaryOut ¶ added in v0.8.0

func AtgMaxUnaryOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgMaxUnpool2d ¶

func AtgMaxUnpool2d(ptr *Ctensor, self Ctensor, indices Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgMaxUnpool2dOut ¶

func AtgMaxUnpool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, indices Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgMaxUnpool3d ¶

func AtgMaxUnpool3d(ptr *Ctensor, self Ctensor, indices Ctensor, outputSizeData []int64, outputSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int)

func AtgMaxUnpool3dOut ¶

func AtgMaxUnpool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, indices Ctensor, outputSizeData []int64, outputSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int)

func AtgMaximum ¶ added in v0.3.0

func AtgMaximum(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgMaximumOut ¶ added in v0.3.0

func AtgMaximumOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgMean ¶

func AtgMean(ptr *Ctensor, self Ctensor, dtype int32)

func AtgMeanDim ¶ added in v0.4.0

func AtgMeanDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgMeanOut ¶

func AtgMeanOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgMedian ¶

func AtgMedian(ptr *Ctensor, self Ctensor)

func AtgMedianDim ¶ added in v0.4.0

func AtgMedianDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgMedianDimValues ¶ added in v0.4.0

func AtgMedianDimValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgMedianOut ¶

func AtgMedianOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgMh ¶ added in v0.7.0

func AtgMh(ptr *Ctensor, self Ctensor)

func AtgMin ¶

func AtgMin(ptr *Ctensor, self Ctensor)

func AtgMinDim ¶ added in v0.4.0

func AtgMinDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgMinDimMin ¶ added in v0.4.0

func AtgMinDimMin(ptr *Ctensor, min Ctensor, minIndices Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgMinOther ¶ added in v0.4.0

func AtgMinOther(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgMinOut ¶

func AtgMinOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgMinUnaryOut ¶ added in v0.9.0

func AtgMinUnaryOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgMinimum ¶ added in v0.3.0

func AtgMinimum(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgMinimumOut ¶ added in v0.3.0

func AtgMinimumOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgMiopenBatchNorm ¶

func AtgMiopenBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, exponentialAverageFactor float64, epsilon float64)

func AtgMiopenBatchNormBackward ¶

func AtgMiopenBatchNormBackward(ptr *Ctensor, input Ctensor, gradOutput Ctensor, weight Ctensor, runningMean Ctensor, runningVar Ctensor, saveMean Ctensor, saveVar Ctensor, epsilon float64)

func AtgMiopenBatchNormBackwardOut ¶ added in v0.8.0

func AtgMiopenBatchNormBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, gradOutput Ctensor, weight Ctensor, runningMean Ctensor, runningVar Ctensor, saveMean Ctensor, saveVar Ctensor, epsilon float64)

func AtgMiopenBatchNormOut ¶ added in v0.8.0

func AtgMiopenBatchNormOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, exponentialAverageFactor float64, epsilon float64)

func AtgMiopenConvolution ¶

func AtgMiopenConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32)

func AtgMiopenConvolutionAddRelu ¶ added in v0.8.0

func AtgMiopenConvolutionAddRelu(ptr *Ctensor, self Ctensor, weight Ctensor, z Ctensor, alpha Cscalar, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, groups int64)

func AtgMiopenConvolutionOut ¶ added in v0.8.0

func AtgMiopenConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32)

func AtgMiopenConvolutionRelu ¶ added in v0.8.0

func AtgMiopenConvolutionRelu(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, groups int64)

func AtgMiopenConvolutionTranspose ¶

func AtgMiopenConvolutionTranspose(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32)

func AtgMiopenConvolutionTransposeOut ¶ added in v0.8.0

func AtgMiopenConvolutionTransposeOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32)

func AtgMiopenDepthwiseConvolution ¶

func AtgMiopenDepthwiseConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32)

func AtgMiopenDepthwiseConvolutionOut ¶ added in v0.8.0

func AtgMiopenDepthwiseConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32)

func AtgMiopenRnn ¶

func AtgMiopenRnn(ptr *Ctensor, input Ctensor, weightData []Ctensor, weightLen int, weightStride0 int64, hx Ctensor, cx Ctensor, mode int64, hiddenSize int64, numLayers int64, batchFirst int32, dropout float64, train int32, bidirectional int32, batchSizesData []int64, batchSizesLen int, dropoutState Ctensor)

func AtgMiopenRnnOut ¶ added in v0.8.0

func AtgMiopenRnnOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, out4 Ctensor, input Ctensor, weightData []Ctensor, weightLen int, weightStride0 int64, hx Ctensor, cx Ctensor, mode int64, hiddenSize int64, numLayers int64, batchFirst int32, dropout float64, train int32, bidirectional int32, batchSizesData []int64, batchSizesLen int, dropoutState Ctensor)

func AtgMish ¶ added in v0.4.0

func AtgMish(ptr *Ctensor, self Ctensor)

func AtgMishBackward ¶ added in v0.4.0

func AtgMishBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)

func AtgMishOut ¶ added in v0.4.0

func AtgMishOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgMish_ ¶ added in v0.4.0

func AtgMish_(ptr *Ctensor, self Ctensor)

func AtgMkldnnAdaptiveAvgPool2d ¶

func AtgMkldnnAdaptiveAvgPool2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgMkldnnAdaptiveAvgPool2dBackward ¶ added in v0.4.0

func AtgMkldnnAdaptiveAvgPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)

func AtgMkldnnAdaptiveAvgPool2dBackwardOut ¶ added in v0.8.0

func AtgMkldnnAdaptiveAvgPool2dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor)

func AtgMkldnnAdaptiveAvgPool2dOut ¶ added in v0.8.0

func AtgMkldnnAdaptiveAvgPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func AtgMkldnnConvolution ¶

func AtgMkldnnConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64)

func AtgMkldnnConvolutionOut ¶ added in v0.8.0

func AtgMkldnnConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64)

func AtgMkldnnLinear ¶

func AtgMkldnnLinear(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor)

func AtgMkldnnLinearBackwardInput ¶ added in v0.4.0

func AtgMkldnnLinearBackwardInput(ptr *Ctensor, inputSizeData []int64, inputSizeLen int, gradOutput Ctensor, weight Ctensor)

func AtgMkldnnLinearBackwardInputOut ¶ added in v0.8.0

func AtgMkldnnLinearBackwardInputOut(ptr *Ctensor, out Ctensor, inputSizeData []int64, inputSizeLen int, gradOutput Ctensor, weight Ctensor)

func AtgMkldnnLinearBackwardWeights ¶ added in v0.4.0

func AtgMkldnnLinearBackwardWeights(ptr *Ctensor, gradOutput Ctensor, input Ctensor, weight Ctensor, biasDefined int32)

func AtgMkldnnLinearBackwardWeightsOut ¶ added in v0.8.0

func AtgMkldnnLinearBackwardWeightsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, gradOutput Ctensor, input Ctensor, weight Ctensor, biasDefined int32)

func AtgMkldnnLinearOut ¶ added in v0.8.0

func AtgMkldnnLinearOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor)

func AtgMkldnnMaxPool2d ¶

func AtgMkldnnMaxPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMkldnnMaxPool2dBackward ¶ added in v0.4.0

func AtgMkldnnMaxPool2dBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor, input Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMkldnnMaxPool2dBackwardOut ¶ added in v0.8.0

func AtgMkldnnMaxPool2dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, input Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMkldnnMaxPool2dOut ¶ added in v0.8.0

func AtgMkldnnMaxPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMkldnnMaxPool3d ¶ added in v0.3.0

func AtgMkldnnMaxPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMkldnnMaxPool3dBackward ¶ added in v0.4.0

func AtgMkldnnMaxPool3dBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor, input Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMkldnnMaxPool3dBackwardOut ¶ added in v0.8.0

func AtgMkldnnMaxPool3dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, input Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMkldnnMaxPool3dOut ¶ added in v0.8.0

func AtgMkldnnMaxPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgMkldnnReorderConv2dWeight ¶

func AtgMkldnnReorderConv2dWeight(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, inputSizeData []int64, inputSizeLen int)

func AtgMkldnnReorderConv2dWeightOut ¶ added in v0.8.0

func AtgMkldnnReorderConv2dWeightOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, inputSizeData []int64, inputSizeLen int)

func AtgMkldnnReorderConv3dWeight ¶ added in v0.3.0

func AtgMkldnnReorderConv3dWeight(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64)

func AtgMkldnnReorderConv3dWeightOut ¶ added in v0.8.0

func AtgMkldnnReorderConv3dWeightOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64)

func AtgMkldnnRnnLayer ¶ added in v0.8.0

func AtgMkldnnRnnLayer(ptr *Ctensor, input Ctensor, weight0 Ctensor, weight1 Ctensor, weight2 Ctensor, weight3 Ctensor, hx_ Ctensor, cx_ Ctensor, reverse int32, batchSizesData []int64, batchSizesLen int, mode int64, hiddenSize int64, numLayers int64, hasBiases int32, bidirectional int32, batchFirst int32, train int32)

func AtgMkldnnRnnLayerBackward ¶ added in v0.8.0

func AtgMkldnnRnnLayerBackward(ptr *Ctensor, input Ctensor, weight1 Ctensor, weight2 Ctensor, weight3 Ctensor, weight4 Ctensor, hx_ Ctensor, cxTmp Ctensor, output Ctensor, hy_ Ctensor, cy_ Ctensor, gradOutput Ctensor, gradHy Ctensor, gradCy Ctensor, reverse int32, mode int64, hiddenSize int64, numLayers int64, hasBiases int32, train int32, bidirectional int32, batchSizesData []int64, batchSizesLen int, batchFirst int32, workspace Ctensor)

func AtgMkldnnRnnLayerBackwardOut ¶ added in v0.8.0

func AtgMkldnnRnnLayerBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, out4 Ctensor, out5 Ctensor, out6 Ctensor, input Ctensor, weight1 Ctensor, weight2 Ctensor, weight3 Ctensor, weight4 Ctensor, hx_ Ctensor, cxTmp Ctensor, output Ctensor, hy_ Ctensor, cy_ Ctensor, gradOutput Ctensor, gradHy Ctensor, gradCy Ctensor, reverse int32, mode int64, hiddenSize int64, numLayers int64, hasBiases int32, train int32, bidirectional int32, batchSizesData []int64, batchSizesLen int, batchFirst int32, workspace Ctensor)

func AtgMkldnnRnnLayerOut ¶ added in v0.8.0

func AtgMkldnnRnnLayerOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, input Ctensor, weight0 Ctensor, weight1 Ctensor, weight2 Ctensor, weight3 Ctensor, hx_ Ctensor, cx_ Ctensor, reverse int32, batchSizesData []int64, batchSizesLen int, mode int64, hiddenSize int64, numLayers int64, hasBiases int32, bidirectional int32, batchFirst int32, train int32)

func AtgMm ¶

func AtgMm(ptr *Ctensor, self Ctensor, mat2 Ctensor)

func AtgMmOut ¶

func AtgMmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor)

func AtgMode ¶

func AtgMode(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgModeValues ¶ added in v0.4.0

func AtgModeValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgMoveaxis ¶ added in v0.4.0

func AtgMoveaxis(ptr *Ctensor, self Ctensor, sourceData []int64, sourceLen int, destinationData []int64, destinationLen int)

func AtgMoveaxisInt ¶ added in v0.4.0

func AtgMoveaxisInt(ptr *Ctensor, self Ctensor, source int64, destination int64)

func AtgMovedim ¶ added in v0.3.0

func AtgMovedim(ptr *Ctensor, self Ctensor, sourceData []int64, sourceLen int, destinationData []int64, destinationLen int)

func AtgMovedimInt ¶ added in v0.4.0

func AtgMovedimInt(ptr *Ctensor, self Ctensor, source int64, destination int64)

func AtgMseLoss ¶

func AtgMseLoss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgMseLossBackward ¶

func AtgMseLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgMseLossBackwardGradInput ¶ added in v0.4.0

func AtgMseLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgMseLossOut ¶

func AtgMseLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgMsort ¶ added in v0.4.0

func AtgMsort(ptr *Ctensor, self Ctensor)

func AtgMsortOut ¶ added in v0.4.0

func AtgMsortOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgMt ¶ added in v0.7.0

func AtgMt(ptr *Ctensor, self Ctensor)

func AtgMul ¶

func AtgMul(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgMulOut ¶

func AtgMulOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgMulScalar ¶ added in v0.4.0

func AtgMulScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgMulScalarOut ¶ added in v0.8.0

func AtgMulScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgMulScalar_ ¶ added in v0.4.0

func AtgMulScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgMul_ ¶

func AtgMul_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgMultiMarginLossBackward ¶

func AtgMultiMarginLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, p Cscalar, margin Cscalar, weight Ctensor, reduction int64)

func AtgMultiMarginLossBackwardGradInput ¶ added in v0.4.0

func AtgMultiMarginLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, p Cscalar, margin Cscalar, weight Ctensor, reduction int64)

func AtgMultilabelMarginLoss ¶

func AtgMultilabelMarginLoss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgMultilabelMarginLossBackward ¶

func AtgMultilabelMarginLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64, isTarget Ctensor)

func AtgMultilabelMarginLossBackwardGradInput ¶ added in v0.4.0

func AtgMultilabelMarginLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64, isTarget Ctensor)

func AtgMultilabelMarginLossOut ¶

func AtgMultilabelMarginLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgMultinomial ¶

func AtgMultinomial(ptr *Ctensor, self Ctensor, numSamples int64, replacement int32)

func AtgMultinomialOut ¶

func AtgMultinomialOut(ptr *Ctensor, out Ctensor, self Ctensor, numSamples int64, replacement int32)

func AtgMultiply ¶ added in v0.3.0

func AtgMultiply(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgMultiplyOut ¶ added in v0.3.0

func AtgMultiplyOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgMultiplyScalar ¶ added in v0.4.0

func AtgMultiplyScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgMultiplyScalar_ ¶ added in v0.4.0

func AtgMultiplyScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgMultiply_ ¶ added in v0.3.0

func AtgMultiply_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgMv ¶

func AtgMv(ptr *Ctensor, self Ctensor, vec Ctensor)

func AtgMvOut ¶

func AtgMvOut(ptr *Ctensor, out Ctensor, self Ctensor, vec Ctensor)

func AtgMvlgamma ¶

func AtgMvlgamma(ptr *Ctensor, self Ctensor, p int64)

func AtgMvlgammaOut ¶ added in v0.5.0

func AtgMvlgammaOut(ptr *Ctensor, out Ctensor, self Ctensor, p int64)

func AtgMvlgamma_ ¶

func AtgMvlgamma_(ptr *Ctensor, self Ctensor, p int64)

func AtgNanToNum ¶ added in v0.4.0

func AtgNanToNum(ptr *Ctensor, self Ctensor, nanVal float64, nanNull int, posinfVal float64, posinfNull int, neginfVal float64, neginfNull int)

func AtgNanToNumOut ¶ added in v0.4.0

func AtgNanToNumOut(ptr *Ctensor, out Ctensor, self Ctensor, nanVal float64, nanNull int, posinfVal float64, posinfNull int, neginfVal float64, neginfNull int)

func AtgNanToNum_ ¶ added in v0.4.0

func AtgNanToNum_(ptr *Ctensor, self Ctensor, nanVal float64, nanNull int, posinfVal float64, posinfNull int, neginfVal float64, neginfNull int)

func AtgNanmean ¶ added in v0.5.0

func AtgNanmean(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgNanmeanOut ¶ added in v0.5.0

func AtgNanmeanOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgNanmedian ¶ added in v0.4.0

func AtgNanmedian(ptr *Ctensor, self Ctensor)

func AtgNanmedianDim ¶ added in v0.4.0

func AtgNanmedianDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgNanmedianDimValues ¶ added in v0.4.0

func AtgNanmedianDimValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64, keepdim int32)

func AtgNanmedianOut ¶ added in v0.8.0

func AtgNanmedianOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgNanquantile ¶ added in v0.3.0

func AtgNanquantile(ptr *Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgNanquantileOut ¶ added in v0.3.0

func AtgNanquantileOut(ptr *Ctensor, out Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgNanquantileScalar ¶ added in v0.4.0

func AtgNanquantileScalar(ptr *Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgNanquantileScalarOut ¶ added in v0.4.0

func AtgNanquantileScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgNansum ¶ added in v0.3.0

func AtgNansum(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgNansumOut ¶ added in v0.3.0

func AtgNansumOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgNarrow ¶

func AtgNarrow(ptr *Ctensor, self Ctensor, dim int64, start int64, length int64)

func AtgNarrowCopy ¶

func AtgNarrowCopy(ptr *Ctensor, self Ctensor, dim int64, start int64, length int64)

func AtgNarrowCopyOut ¶ added in v0.4.0

func AtgNarrowCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, start int64, length int64)

func AtgNarrowTensor ¶ added in v0.4.0

func AtgNarrowTensor(ptr *Ctensor, self Ctensor, dim int64, start Ctensor, length int64)

func AtgNativeBatchNorm ¶

func AtgNativeBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, momentum float64, eps float64)

func AtgNativeBatchNormOut ¶

func AtgNativeBatchNormOut(ptr *Ctensor, out Ctensor, saveMean Ctensor, saveInvstd Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, momentum float64, eps float64)

func AtgNativeChannelShuffle ¶ added in v0.7.0

func AtgNativeChannelShuffle(ptr *Ctensor, self Ctensor, groups int64)

func AtgNativeDropout ¶ added in v0.7.0

func AtgNativeDropout(ptr *Ctensor, input Ctensor, p float64, train int32)

func AtgNativeDropoutBackward ¶ added in v0.7.0

func AtgNativeDropoutBackward(ptr *Ctensor, gradOutput Ctensor, mask Ctensor, scale float64)

func AtgNativeDropoutBackwardOut ¶ added in v0.8.0

func AtgNativeDropoutBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, mask Ctensor, scale float64)

func AtgNativeDropoutOut ¶ added in v0.8.0

func AtgNativeDropoutOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, p float64, train int32)

func AtgNativeGroupNorm ¶ added in v0.3.0

func AtgNativeGroupNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, n int64, c int64, hxW int64, group int64, eps float64)

func AtgNativeGroupNormOut ¶ added in v0.8.0

func AtgNativeGroupNormOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, weight Ctensor, bias Ctensor, n int64, c int64, hxW int64, group int64, eps float64)

func AtgNativeLayerNorm ¶

func AtgNativeLayerNorm(ptr *Ctensor, input Ctensor, normalizedShapeData []int64, normalizedShapeLen int, weight Ctensor, bias Ctensor, eps float64)

func AtgNativeLayerNormOut ¶ added in v0.8.0

func AtgNativeLayerNormOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, normalizedShapeData []int64, normalizedShapeLen int, weight Ctensor, bias Ctensor, eps float64)

func AtgNativeNorm ¶

func AtgNativeNorm(ptr *Ctensor, self Ctensor)

func AtgNativeNormOut ¶ added in v0.8.0

func AtgNativeNormOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgNativeNormScalaroptDimDtype ¶ added in v0.4.0

func AtgNativeNormScalaroptDimDtype(ptr *Ctensor, self Ctensor, p Cscalar, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgNativeNormScalaroptDimDtypeOut ¶ added in v0.8.0

func AtgNativeNormScalaroptDimDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgNe ¶

func AtgNe(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgNeScalarOut ¶ added in v0.4.0

func AtgNeScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgNeTensor ¶ added in v0.4.0

func AtgNeTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgNeTensorOut ¶ added in v0.4.0

func AtgNeTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgNeTensor_ ¶ added in v0.4.0

func AtgNeTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgNe_ ¶

func AtgNe_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgNeg ¶

func AtgNeg(ptr *Ctensor, self Ctensor)

func AtgNegOut ¶

func AtgNegOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgNeg_ ¶

func AtgNeg_(ptr *Ctensor, self Ctensor)

func AtgNegative ¶ added in v0.3.0

func AtgNegative(ptr *Ctensor, self Ctensor)

func AtgNegativeOut ¶ added in v0.3.0

func AtgNegativeOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgNegative_ ¶ added in v0.3.0

func AtgNegative_(ptr *Ctensor, self Ctensor)

func AtgNestedToPaddedTensor ¶ added in v0.8.0

func AtgNestedToPaddedTensor(ptr *Ctensor, self Ctensor, padding float64, outputSizeData []int64, outputSizeLen int)

func AtgNewEmpty ¶

func AtgNewEmpty(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgNewEmptyOut ¶ added in v0.8.0

func AtgNewEmptyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgNewEmptyStrided ¶ added in v0.4.0

func AtgNewEmptyStrided(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int, optionsKind int32, optionsDevice int32)

func AtgNewEmptyStridedOut ¶ added in v0.8.0

func AtgNewEmptyStridedOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int)

func AtgNewFull ¶

func AtgNewFull(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, fillValue Cscalar, optionsKind int32, optionsDevice int32)

func AtgNewFullOut ¶ added in v0.8.0

func AtgNewFullOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, fillValue Cscalar)

func AtgNewOnes ¶ added in v0.5.0

func AtgNewOnes(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgNewOnesOut ¶ added in v0.8.0

func AtgNewOnesOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgNewZeros ¶

func AtgNewZeros(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgNewZerosOut ¶ added in v0.8.0

func AtgNewZerosOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgNextafter ¶ added in v0.3.0

func AtgNextafter(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgNextafterOut ¶ added in v0.3.0

func AtgNextafterOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgNextafter_ ¶ added in v0.3.0

func AtgNextafter_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgNllLoss ¶

func AtgNllLoss(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64)

func AtgNllLoss2d ¶

func AtgNllLoss2d(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64)

func AtgNllLoss2dBackward ¶

func AtgNllLoss2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64, totalWeight Ctensor)

func AtgNllLoss2dBackwardGradInput ¶ added in v0.4.0

func AtgNllLoss2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64, totalWeight Ctensor)

func AtgNllLoss2dOut ¶

func AtgNllLoss2dOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64)

func AtgNllLossBackward ¶

func AtgNllLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64, totalWeight Ctensor)

func AtgNllLossBackwardGradInput ¶ added in v0.4.0

func AtgNllLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64, totalWeight Ctensor)

func AtgNllLossNd ¶ added in v0.4.0

func AtgNllLossNd(ptr *Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64)

func AtgNllLossOut ¶

func AtgNllLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, weight Ctensor, reduction int64, ignoreIndex int64)

func AtgNonzero ¶

func AtgNonzero(ptr *Ctensor, self Ctensor)

func AtgNonzeroOut ¶

func AtgNonzeroOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgNonzeroStatic ¶ added in v0.9.0

func AtgNonzeroStatic(ptr *Ctensor, self Ctensor, size int64, fillValue int64)

func AtgNonzeroStaticOut ¶ added in v0.9.0

func AtgNonzeroStaticOut(ptr *Ctensor, out Ctensor, self Ctensor, size int64, fillValue int64)

func AtgNorm ¶

func AtgNorm(ptr *Ctensor, self Ctensor)

func AtgNormDtypeOut ¶ added in v0.4.0

func AtgNormDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgNormExceptDim ¶

func AtgNormExceptDim(ptr *Ctensor, v Ctensor, pow int64, dim int64)

func AtgNormOut ¶

func AtgNormOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dimData []int64, dimLen int, keepdim int32)

func AtgNormScalarOut ¶ added in v0.8.0

func AtgNormScalarOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgNormScalaroptDim ¶ added in v0.4.0

func AtgNormScalaroptDim(ptr *Ctensor, self Ctensor, p Cscalar, dimData []int64, dimLen int, keepdim int32)

func AtgNormScalaroptDimDtype ¶ added in v0.4.0

func AtgNormScalaroptDimDtype(ptr *Ctensor, self Ctensor, p Cscalar, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgNormScalaroptDtype ¶ added in v0.4.0

func AtgNormScalaroptDtype(ptr *Ctensor, self Ctensor, p Cscalar, dtype int32)

func AtgNormScalaroptDtypeOut ¶ added in v0.8.0

func AtgNormScalaroptDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dtype int32)

func AtgNormalFunctional ¶ added in v0.8.0

func AtgNormalFunctional(ptr *Ctensor, self Ctensor, mean float64, std float64)

func AtgNormal_ ¶

func AtgNormal_(ptr *Ctensor, self Ctensor, mean float64, std float64)

func AtgNotEqual ¶ added in v0.3.0

func AtgNotEqual(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgNotEqualScalarOut ¶ added in v0.4.0

func AtgNotEqualScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgNotEqualTensor ¶ added in v0.4.0

func AtgNotEqualTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgNotEqualTensorOut ¶ added in v0.4.0

func AtgNotEqualTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgNotEqualTensor_ ¶ added in v0.4.0

func AtgNotEqualTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgNotEqual_ ¶ added in v0.3.0

func AtgNotEqual_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgNuclearNorm ¶

func AtgNuclearNorm(ptr *Ctensor, self Ctensor, keepdim int32)

func AtgNuclearNormDim ¶ added in v0.4.0

func AtgNuclearNormDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgNuclearNormDimOut ¶ added in v0.4.0

func AtgNuclearNormDimOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgNuclearNormOut ¶

func AtgNuclearNormOut(ptr *Ctensor, out Ctensor, self Ctensor, keepdim int32)

func AtgNumpyT ¶

func AtgNumpyT(ptr *Ctensor, self Ctensor)

func AtgOneHot ¶

func AtgOneHot(ptr *Ctensor, self Ctensor, numClasses int64)

func AtgOnes ¶

func AtgOnes(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgOnesLike ¶

func AtgOnesLike(ptr *Ctensor, self Ctensor)

func AtgOnesLikeOut ¶ added in v0.8.0

func AtgOnesLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgOnesOut ¶

func AtgOnesOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)

func AtgOrgqr ¶

func AtgOrgqr(ptr *Ctensor, self Ctensor, input2 Ctensor)

func AtgOrgqrOut ¶

func AtgOrgqrOut(ptr *Ctensor, out Ctensor, self Ctensor, input2 Ctensor)

func AtgOrmqr ¶

func AtgOrmqr(ptr *Ctensor, self Ctensor, input2 Ctensor, input3 Ctensor, left int32, transpose int32)

func AtgOrmqrOut ¶

func AtgOrmqrOut(ptr *Ctensor, out Ctensor, self Ctensor, input2 Ctensor, input3 Ctensor, left int32, transpose int32)

func AtgOuter ¶ added in v0.3.0

func AtgOuter(ptr *Ctensor, self Ctensor, vec2 Ctensor)

func AtgOuterOut ¶ added in v0.3.0

func AtgOuterOut(ptr *Ctensor, out Ctensor, self Ctensor, vec2 Ctensor)

func AtgOutputNr ¶ added in v0.4.0

func AtgOutputNr(self Ctensor) int64

func AtgPad ¶ added in v0.8.0

func AtgPad(ptr *Ctensor, self Ctensor, padData []int64, padLen int, mode string, valueVal float64, valueNull int)

func AtgPadSequence ¶ added in v0.4.0

func AtgPadSequence(ptr *Ctensor, sequencesData []Ctensor, sequencesLen int, batchFirst int32, paddingValue float64)

func AtgPairwiseDistance ¶

func AtgPairwiseDistance(ptr *Ctensor, x1 Ctensor, x2 Ctensor, p float64, eps float64, keepdim int32)

func AtgPdist ¶

func AtgPdist(ptr *Ctensor, self Ctensor, p float64)

func AtgPermute ¶

func AtgPermute(ptr *Ctensor, self Ctensor, dimsData []int64, dimsLen int)

func AtgPermuteCopy ¶ added in v0.8.0

func AtgPermuteCopy(ptr *Ctensor, self Ctensor, dimsData []int64, dimsLen int)

func AtgPermuteCopyOut ¶ added in v0.8.0

func AtgPermuteCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dimsData []int64, dimsLen int)

func AtgPinMemory ¶

func AtgPinMemory(ptr *Ctensor, self Ctensor, device int32)

func AtgPinverse ¶

func AtgPinverse(ptr *Ctensor, self Ctensor, rcond float64)

func AtgPixelShuffle ¶

func AtgPixelShuffle(ptr *Ctensor, self Ctensor, upscaleFactor int64)

func AtgPixelShuffleOut ¶ added in v0.8.0

func AtgPixelShuffleOut(ptr *Ctensor, out Ctensor, self Ctensor, upscaleFactor int64)

func AtgPixelUnshuffle ¶ added in v0.4.0

func AtgPixelUnshuffle(ptr *Ctensor, self Ctensor, downscaleFactor int64)

func AtgPixelUnshuffleOut ¶ added in v0.8.0

func AtgPixelUnshuffleOut(ptr *Ctensor, out Ctensor, self Ctensor, downscaleFactor int64)

func AtgPoisson ¶

func AtgPoisson(ptr *Ctensor, self Ctensor)

func AtgPoissonNllLoss ¶

func AtgPoissonNllLoss(ptr *Ctensor, input Ctensor, target Ctensor, logInput int32, full int32, eps float64, reduction int64)

func AtgPoissonOut ¶ added in v0.8.0

func AtgPoissonOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgPolar ¶ added in v0.3.0

func AtgPolar(ptr *Ctensor, abs Ctensor, angle Ctensor)

func AtgPolarOut ¶ added in v0.3.0

func AtgPolarOut(ptr *Ctensor, out Ctensor, abs Ctensor, angle Ctensor)

func AtgPolygamma ¶

func AtgPolygamma(ptr *Ctensor, n int64, self Ctensor)

func AtgPolygammaOut ¶

func AtgPolygammaOut(ptr *Ctensor, out Ctensor, n int64, self Ctensor)

func AtgPolygamma_ ¶

func AtgPolygamma_(ptr *Ctensor, self Ctensor, n int64)

func AtgPositive ¶ added in v0.4.0

func AtgPositive(ptr *Ctensor, self Ctensor)

func AtgPow ¶

func AtgPow(ptr *Ctensor, self Ctensor, exponent Ctensor)

func AtgPowScalar ¶ added in v0.4.0

func AtgPowScalar(ptr *Ctensor, selfScalar Cscalar, exponent Ctensor)

func AtgPowScalarOut ¶ added in v0.4.0

func AtgPowScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, exponent Ctensor)

func AtgPowTensorScalar ¶ added in v0.5.0

func AtgPowTensorScalar(ptr *Ctensor, self Ctensor, exponent Cscalar)

func AtgPowTensorScalarOut ¶ added in v0.4.0

func AtgPowTensorScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, exponent Cscalar)

func AtgPowTensorTensorOut ¶ added in v0.4.0

func AtgPowTensorTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, exponent Ctensor)

func AtgPowTensor_ ¶ added in v0.4.0

func AtgPowTensor_(ptr *Ctensor, self Ctensor, exponent Ctensor)

func AtgPow_ ¶

func AtgPow_(ptr *Ctensor, self Ctensor, exponent Cscalar)

func AtgPrelu ¶

func AtgPrelu(ptr *Ctensor, self Ctensor, weight Ctensor)

func AtgProd ¶

func AtgProd(ptr *Ctensor, self Ctensor, dtype int32)

func AtgProdDimInt ¶ added in v0.4.0

func AtgProdDimInt(ptr *Ctensor, self Ctensor, dim int64, keepdim int32, dtype int32)

func AtgProdIntOut ¶ added in v0.4.0

func AtgProdIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, keepdim int32, dtype int32)

func AtgProdOut ¶

func AtgProdOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32)

func AtgPut ¶ added in v0.4.0

func AtgPut(ptr *Ctensor, self Ctensor, index Ctensor, source Ctensor, accumulate int32)

func AtgPutOut ¶ added in v0.8.0

func AtgPutOut(ptr *Ctensor, out Ctensor, self Ctensor, index Ctensor, source Ctensor, accumulate int32)

func AtgPut_ ¶

func AtgPut_(ptr *Ctensor, self Ctensor, index Ctensor, source Ctensor, accumulate int32)

func AtgQPerChannelAxis ¶ added in v0.4.0

func AtgQPerChannelAxis(self Ctensor) int64

func AtgQPerChannelScales ¶

func AtgQPerChannelScales(ptr *Ctensor, self Ctensor)

func AtgQPerChannelScalesOut ¶ added in v0.8.0

func AtgQPerChannelScalesOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgQPerChannelZeroPoints ¶

func AtgQPerChannelZeroPoints(ptr *Ctensor, self Ctensor)

func AtgQPerChannelZeroPointsOut ¶ added in v0.8.0

func AtgQPerChannelZeroPointsOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgQScale ¶ added in v0.4.0

func AtgQScale(self Ctensor) float64

func AtgQZeroPoint ¶ added in v0.4.0

func AtgQZeroPoint(self Ctensor) int64

func AtgQr ¶

func AtgQr(ptr *Ctensor, self Ctensor, some int32)

func AtgQrQ ¶ added in v0.4.0

func AtgQrQ(ptr *Ctensor, q Ctensor, r Ctensor, self Ctensor, some int32)

func AtgQuantile ¶ added in v0.3.0

func AtgQuantile(ptr *Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgQuantileOut ¶ added in v0.3.0

func AtgQuantileOut(ptr *Ctensor, out Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgQuantileScalar ¶ added in v0.4.0

func AtgQuantileScalar(ptr *Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgQuantileScalarOut ¶ added in v0.4.0

func AtgQuantileScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgQuantizePerChannel ¶

func AtgQuantizePerChannel(ptr *Ctensor, self Ctensor, scales Ctensor, zeroPoints Ctensor, axis int64, dtype int32)

func AtgQuantizePerChannelOut ¶ added in v0.8.0

func AtgQuantizePerChannelOut(ptr *Ctensor, out Ctensor, self Ctensor, scales Ctensor, zeroPoints Ctensor, axis int64, dtype int32)

func AtgQuantizePerTensor ¶

func AtgQuantizePerTensor(ptr *Ctensor, self Ctensor, scale float64, zeroPoint int64, dtype int32)

func AtgQuantizePerTensorDynamic ¶ added in v0.7.0

func AtgQuantizePerTensorDynamic(ptr *Ctensor, self Ctensor, dtype int32, reduceRange int32)

func AtgQuantizePerTensorDynamicOut ¶ added in v0.8.0

func AtgQuantizePerTensorDynamicOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32, reduceRange int32)

func AtgQuantizePerTensorTensorQparams ¶ added in v0.5.0

func AtgQuantizePerTensorTensorQparams(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, dtype int32)

func AtgQuantizedBatchNorm ¶

func AtgQuantizedBatchNorm(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, mean Ctensor, vari Ctensor, eps float64, outputScale float64, outputZeroPoint int64)

func AtgQuantizedBatchNormOut ¶ added in v0.8.0

func AtgQuantizedBatchNormOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, mean Ctensor, vari Ctensor, eps float64, outputScale float64, outputZeroPoint int64)

func AtgQuantizedGruCell ¶

func AtgQuantizedGruCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, bHh Ctensor, packedIh Ctensor, packedHh Ctensor, colOffsetsIh Ctensor, colOffsetsHh Ctensor, scaleIh Cscalar, scaleHh Cscalar, zeroPointIh Cscalar, zeroPointHh Cscalar)

func AtgQuantizedLstmCell ¶

func AtgQuantizedLstmCell(ptr *Ctensor, input Ctensor, hxData []Ctensor, hxLen int, wIh Ctensor, wHh Ctensor, bIh Ctensor, bHh Ctensor, packedIh Ctensor, packedHh Ctensor, colOffsetsIh Ctensor, colOffsetsHh Ctensor, scaleIh Cscalar, scaleHh Cscalar, zeroPointIh Cscalar, zeroPointHh Cscalar)

func AtgQuantizedMaxPool1d ¶ added in v0.3.0

func AtgQuantizedMaxPool1d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgQuantizedMaxPool1dOut ¶ added in v0.8.0

func AtgQuantizedMaxPool1dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgQuantizedMaxPool2d ¶

func AtgQuantizedMaxPool2d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgQuantizedMaxPool2dOut ¶ added in v0.8.0

func AtgQuantizedMaxPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgQuantizedMaxPool3d ¶ added in v0.9.0

func AtgQuantizedMaxPool3d(ptr *Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgQuantizedMaxPool3dOut ¶ added in v0.9.0

func AtgQuantizedMaxPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, ceilMode int32)

func AtgQuantizedRnnReluCell ¶

func AtgQuantizedRnnReluCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, bHh Ctensor, packedIh Ctensor, packedHh Ctensor, colOffsetsIh Ctensor, colOffsetsHh Ctensor, scaleIh Cscalar, scaleHh Cscalar, zeroPointIh Cscalar, zeroPointHh Cscalar)

func AtgQuantizedRnnTanhCell ¶

func AtgQuantizedRnnTanhCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, bHh Ctensor, packedIh Ctensor, packedHh Ctensor, colOffsetsIh Ctensor, colOffsetsHh Ctensor, scaleIh Cscalar, scaleHh Cscalar, zeroPointIh Cscalar, zeroPointHh Cscalar)

func AtgRad2deg ¶ added in v0.3.0

func AtgRad2deg(ptr *Ctensor, self Ctensor)

func AtgRad2degOut ¶ added in v0.3.0

func AtgRad2degOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgRad2deg_ ¶ added in v0.3.0

func AtgRad2deg_(ptr *Ctensor, self Ctensor)

func AtgRand ¶

func AtgRand(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgRandLike ¶

func AtgRandLike(ptr *Ctensor, self Ctensor)

func AtgRandLikeOut ¶ added in v0.8.0

func AtgRandLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgRandOut ¶

func AtgRandOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)

func AtgRandint ¶

func AtgRandint(ptr *Ctensor, high int64, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgRandintLike ¶

func AtgRandintLike(ptr *Ctensor, self Ctensor, high int64)

func AtgRandintLikeLowDtype ¶ added in v0.4.0

func AtgRandintLikeLowDtype(ptr *Ctensor, self Ctensor, low int64, high int64)

func AtgRandintLikeLowDtypeOut ¶ added in v0.8.0

func AtgRandintLikeLowDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, low int64, high int64)

func AtgRandintLikeOut ¶ added in v0.8.0

func AtgRandintLikeOut(ptr *Ctensor, out Ctensor, self Ctensor, high int64)

func AtgRandintLow ¶ added in v0.4.0

func AtgRandintLow(ptr *Ctensor, low int64, high int64, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgRandintLowOut ¶ added in v0.4.0

func AtgRandintLowOut(ptr *Ctensor, out Ctensor, low int64, high int64, sizeData []int64, sizeLen int)

func AtgRandintOut ¶

func AtgRandintOut(ptr *Ctensor, out Ctensor, high int64, sizeData []int64, sizeLen int)

func AtgRandn ¶

func AtgRandn(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgRandnLike ¶

func AtgRandnLike(ptr *Ctensor, self Ctensor)

func AtgRandnLikeOut ¶ added in v0.8.0

func AtgRandnLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgRandnOut ¶

func AtgRandnOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)

func AtgRandom ¶ added in v0.8.0

func AtgRandom(ptr *Ctensor, self Ctensor)

func AtgRandomFrom ¶ added in v0.8.0

func AtgRandomFrom(ptr *Ctensor, self Ctensor, from int64, toVal int64, toNull int)

func AtgRandomFromOut ¶ added in v0.8.0

func AtgRandomFromOut(ptr *Ctensor, out Ctensor, self Ctensor, from int64, toVal int64, toNull int)

func AtgRandomFrom_ ¶ added in v0.4.0

func AtgRandomFrom_(ptr *Ctensor, self Ctensor, from int64, toVal int64, toNull int)

func AtgRandomOut ¶ added in v0.8.0

func AtgRandomOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgRandomTo ¶ added in v0.8.0

func AtgRandomTo(ptr *Ctensor, self Ctensor, to int64)

func AtgRandomToOut ¶ added in v0.8.0

func AtgRandomToOut(ptr *Ctensor, out Ctensor, self Ctensor, to int64)

func AtgRandomTo_ ¶ added in v0.4.0

func AtgRandomTo_(ptr *Ctensor, self Ctensor, to int64)

func AtgRandom_ ¶

func AtgRandom_(ptr *Ctensor, self Ctensor)

func AtgRandperm ¶

func AtgRandperm(ptr *Ctensor, n int64, optionsKind int32, optionsDevice int32)

func AtgRandpermOut ¶

func AtgRandpermOut(ptr *Ctensor, out Ctensor, n int64)

func AtgRange ¶

func AtgRange(ptr *Ctensor, start Cscalar, end Cscalar, optionsKind int32, optionsDevice int32)

func AtgRangeOut ¶

func AtgRangeOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar)

func AtgRangeOut_ ¶ added in v0.8.0

func AtgRangeOut_(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar)

func AtgRangeStep ¶ added in v0.4.0

func AtgRangeStep(ptr *Ctensor, start Cscalar, end Cscalar, optionsKind int32, optionsDevice int32)

func AtgRavel ¶ added in v0.4.0

func AtgRavel(ptr *Ctensor, self Ctensor)

func AtgReal ¶

func AtgReal(ptr *Ctensor, self Ctensor)

func AtgReciprocal ¶

func AtgReciprocal(ptr *Ctensor, self Ctensor)

func AtgReciprocalOut ¶

func AtgReciprocalOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgReciprocal_ ¶

func AtgReciprocal_(ptr *Ctensor, self Ctensor)

func AtgReflectionPad1d ¶

func AtgReflectionPad1d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad1dBackward ¶

func AtgReflectionPad1dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad1dBackwardGradInput ¶ added in v0.4.0

func AtgReflectionPad1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad1dOut ¶

func AtgReflectionPad1dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad2d ¶

func AtgReflectionPad2d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad2dBackward ¶

func AtgReflectionPad2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad2dBackwardGradInput ¶ added in v0.4.0

func AtgReflectionPad2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad2dOut ¶

func AtgReflectionPad2dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad3d ¶ added in v0.5.0

func AtgReflectionPad3d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad3dBackward ¶ added in v0.5.0

func AtgReflectionPad3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad3dBackwardGradInput ¶ added in v0.5.0

func AtgReflectionPad3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReflectionPad3dOut ¶ added in v0.5.0

func AtgReflectionPad3dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgRelu ¶

func AtgRelu(ptr *Ctensor, self Ctensor)

func AtgRelu6 ¶ added in v0.4.0

func AtgRelu6(ptr *Ctensor, self Ctensor)

func AtgRelu6_ ¶ added in v0.4.0

func AtgRelu6_(ptr *Ctensor, self Ctensor)

func AtgReluOut ¶ added in v0.8.0

func AtgReluOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgRelu_ ¶

func AtgRelu_(ptr *Ctensor, self Ctensor)

func AtgRemainder ¶

func AtgRemainder(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgRemainderScalarOut ¶ added in v0.4.0

func AtgRemainderScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgRemainderScalarTensor ¶ added in v0.5.0

func AtgRemainderScalarTensor(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgRemainderScalarTensorOut ¶ added in v0.8.0

func AtgRemainderScalarTensorOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgRemainderTensor ¶ added in v0.4.0

func AtgRemainderTensor(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgRemainderTensorOut ¶ added in v0.4.0

func AtgRemainderTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgRemainderTensor_ ¶ added in v0.4.0

func AtgRemainderTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgRemainder_ ¶

func AtgRemainder_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgRenorm ¶

func AtgRenorm(ptr *Ctensor, self Ctensor, p Cscalar, dim int64, maxnorm Cscalar)

func AtgRenormOut ¶

func AtgRenormOut(ptr *Ctensor, out Ctensor, self Ctensor, p Cscalar, dim int64, maxnorm Cscalar)

func AtgRenorm_ ¶

func AtgRenorm_(ptr *Ctensor, self Ctensor, p Cscalar, dim int64, maxnorm Cscalar)

func AtgRepeat ¶

func AtgRepeat(ptr *Ctensor, self Ctensor, repeatsData []int64, repeatsLen int)

func AtgRepeatInterleave ¶

func AtgRepeatInterleave(ptr *Ctensor, repeats Ctensor, outputSizeVal int64, outputSizeNull int)

func AtgRepeatInterleaveSelfInt ¶ added in v0.4.0

func AtgRepeatInterleaveSelfInt(ptr *Ctensor, self Ctensor, repeats int64, dimVal int64, dimNull int, outputSizeVal int64, outputSizeNull int)

func AtgRepeatInterleaveSelfTensor ¶ added in v0.4.0

func AtgRepeatInterleaveSelfTensor(ptr *Ctensor, self Ctensor, repeats Ctensor, dimVal int64, dimNull int, outputSizeVal int64, outputSizeNull int)

func AtgRepeatInterleaveTensorOut ¶ added in v0.8.0

func AtgRepeatInterleaveTensorOut(ptr *Ctensor, out Ctensor, repeats Ctensor, outputSizeVal int64, outputSizeNull int)

func AtgRepeatOut ¶ added in v0.8.0

func AtgRepeatOut(ptr *Ctensor, out Ctensor, self Ctensor, repeatsData []int64, repeatsLen int)

func AtgReplicationPad1d ¶

func AtgReplicationPad1d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad1dBackward ¶

func AtgReplicationPad1dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad1dBackwardGradInput ¶ added in v0.4.0

func AtgReplicationPad1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad1dOut ¶

func AtgReplicationPad1dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad2d ¶

func AtgReplicationPad2d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad2dBackward ¶

func AtgReplicationPad2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad2dBackwardGradInput ¶ added in v0.4.0

func AtgReplicationPad2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad2dOut ¶

func AtgReplicationPad2dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad3d ¶

func AtgReplicationPad3d(ptr *Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad3dBackward ¶

func AtgReplicationPad3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad3dBackwardGradInput ¶ added in v0.4.0

func AtgReplicationPad3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgReplicationPad3dOut ¶

func AtgReplicationPad3dOut(ptr *Ctensor, out Ctensor, self Ctensor, paddingData []int64, paddingLen int)

func AtgRequiresGrad_ ¶

func AtgRequiresGrad_(ptr *Ctensor, self Ctensor, requiresGrad int32)

func AtgReshape ¶

func AtgReshape(ptr *Ctensor, self Ctensor, shapeData []int64, shapeLen int)

func AtgReshapeAs ¶

func AtgReshapeAs(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgResize ¶ added in v0.8.0

func AtgResize(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgResizeAs ¶ added in v0.8.0

func AtgResizeAs(ptr *Ctensor, self Ctensor, theTemplate Ctensor)

func AtgResizeAsOut ¶ added in v0.8.0

func AtgResizeAsOut(ptr *Ctensor, out Ctensor, self Ctensor, theTemplate Ctensor)

func AtgResizeAsSparse ¶ added in v0.8.0

func AtgResizeAsSparse(ptr *Ctensor, self Ctensor, theTemplate Ctensor)

func AtgResizeAsSparseOut ¶ added in v0.8.0

func AtgResizeAsSparseOut(ptr *Ctensor, out Ctensor, self Ctensor, theTemplate Ctensor)

func AtgResizeAsSparse_ ¶ added in v0.4.0

func AtgResizeAsSparse_(ptr *Ctensor, self Ctensor, theTemplate Ctensor)

func AtgResizeAs_ ¶

func AtgResizeAs_(ptr *Ctensor, self Ctensor, theTemplate Ctensor)

func AtgResizeOut ¶ added in v0.8.0

func AtgResizeOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgResize_ ¶

func AtgResize_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgResolveConj ¶ added in v0.5.0

func AtgResolveConj(ptr *Ctensor, self Ctensor)

func AtgResolveNeg ¶ added in v0.5.0

func AtgResolveNeg(ptr *Ctensor, self Ctensor)

func AtgRetainsGrad ¶ added in v0.5.0

func AtgRetainsGrad(self Ctensor) bool

func AtgRnnRelu ¶

func AtgRnnRelu(ptr *Ctensor, input Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32, batchFirst int32)

func AtgRnnReluCell ¶

func AtgRnnReluCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, bHh Ctensor)

func AtgRnnReluData ¶ added in v0.4.0

func AtgRnnReluData(ptr *Ctensor, data Ctensor, batchSizes Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32)

func AtgRnnTanh ¶

func AtgRnnTanh(ptr *Ctensor, input Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32, batchFirst int32)

func AtgRnnTanhCell ¶

func AtgRnnTanhCell(ptr *Ctensor, input Ctensor, hx Ctensor, wIh Ctensor, wHh Ctensor, bIh Ctensor, bHh Ctensor)

func AtgRnnTanhData ¶ added in v0.4.0

func AtgRnnTanhData(ptr *Ctensor, data Ctensor, batchSizes Ctensor, hx Ctensor, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32)

func AtgRoll ¶

func AtgRoll(ptr *Ctensor, self Ctensor, shiftsData []int64, shiftsLen int, dimsData []int64, dimsLen int)

func AtgRollOut ¶ added in v0.8.0

func AtgRollOut(ptr *Ctensor, out Ctensor, self Ctensor, shiftsData []int64, shiftsLen int, dimsData []int64, dimsLen int)

func AtgRot90 ¶

func AtgRot90(ptr *Ctensor, self Ctensor, k int64, dimsData []int64, dimsLen int)

func AtgRot90Out ¶ added in v0.8.0

func AtgRot90Out(ptr *Ctensor, out Ctensor, self Ctensor, k int64, dimsData []int64, dimsLen int)

func AtgRound ¶

func AtgRound(ptr *Ctensor, self Ctensor)

func AtgRoundDecimals ¶ added in v0.7.0

func AtgRoundDecimals(ptr *Ctensor, self Ctensor, decimals int64)

func AtgRoundDecimalsOut ¶ added in v0.7.0

func AtgRoundDecimalsOut(ptr *Ctensor, out Ctensor, self Ctensor, decimals int64)

func AtgRoundDecimals_ ¶ added in v0.7.0

func AtgRoundDecimals_(ptr *Ctensor, self Ctensor, decimals int64)

func AtgRoundOut ¶

func AtgRoundOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgRound_ ¶

func AtgRound_(ptr *Ctensor, self Ctensor)

func AtgRowIndices ¶ added in v0.8.0

func AtgRowIndices(ptr *Ctensor, self Ctensor)

func AtgRowIndicesCopy ¶ added in v0.8.0

func AtgRowIndicesCopy(ptr *Ctensor, self Ctensor)

func AtgRowIndicesCopyOut ¶ added in v0.8.0

func AtgRowIndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgRowStack ¶ added in v0.4.0

func AtgRowStack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgRowStackOut ¶ added in v0.4.0

func AtgRowStackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgRrelu ¶

func AtgRrelu(ptr *Ctensor, self Ctensor, training int32)

func AtgRreluWithNoise ¶

func AtgRreluWithNoise(ptr *Ctensor, self Ctensor, noise Ctensor, training int32)

func AtgRreluWithNoiseBackward ¶

func AtgRreluWithNoiseBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, noise Ctensor, lower Cscalar, upper Cscalar, training int32, selfIsResult int32)

func AtgRreluWithNoiseBackwardOut ¶ added in v0.8.0

func AtgRreluWithNoiseBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor, noise Ctensor, lower Cscalar, upper Cscalar, training int32, selfIsResult int32)

func AtgRreluWithNoiseOut ¶

func AtgRreluWithNoiseOut(ptr *Ctensor, out Ctensor, self Ctensor, noise Ctensor, training int32)

func AtgRreluWithNoise_ ¶

func AtgRreluWithNoise_(ptr *Ctensor, self Ctensor, noise Ctensor, training int32)

func AtgRrelu_ ¶

func AtgRrelu_(ptr *Ctensor, self Ctensor, training int32)

func AtgRsqrt ¶

func AtgRsqrt(ptr *Ctensor, self Ctensor)

func AtgRsqrtOut ¶

func AtgRsqrtOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgRsqrt_ ¶

func AtgRsqrt_(ptr *Ctensor, self Ctensor)

func AtgRsub ¶

func AtgRsub(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgRsubScalar ¶ added in v0.4.0

func AtgRsubScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgRsubScalarOut ¶ added in v0.8.0

func AtgRsubScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgRsubTensorOut ¶ added in v0.8.0

func AtgRsubTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgScalarTensor ¶

func AtgScalarTensor(ptr *Ctensor, s Cscalar, optionsKind int32, optionsDevice int32)

func AtgScalarTensorOut ¶ added in v0.8.0

func AtgScalarTensorOut(ptr *Ctensor, out Ctensor, s Cscalar)

func AtgScaledDotProductAttention ¶ added in v0.8.0

func AtgScaledDotProductAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int)

func AtgScatter ¶

func AtgScatter(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)

func AtgScatterAdd ¶

func AtgScatterAdd(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)

func AtgScatterAddOut ¶ added in v0.5.0

func AtgScatterAddOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)

func AtgScatterAdd_ ¶

func AtgScatterAdd_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)

func AtgScatterReduce ¶ added in v0.5.0

func AtgScatterReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, reduce string)

func AtgScatterReduceOut ¶ added in v0.5.0

func AtgScatterReduceOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, reduce string)

func AtgScatterReduce_ ¶ added in v0.4.0

func AtgScatterReduce_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, reduce string)

func AtgScatterSrcOut ¶ added in v0.5.0

func AtgScatterSrcOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)

func AtgScatterValue ¶ added in v0.4.0

func AtgScatterValue(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)

func AtgScatterValueOut ¶ added in v0.5.0

func AtgScatterValueOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)

func AtgScatterValueReduce ¶ added in v0.5.0

func AtgScatterValueReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar, reduce string)

func AtgScatterValueReduceOut ¶ added in v0.5.0

func AtgScatterValueReduceOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar, reduce string)

func AtgScatterValueReduce_ ¶ added in v0.4.0

func AtgScatterValueReduce_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar, reduce string)

func AtgScatterValue_ ¶ added in v0.4.0

func AtgScatterValue_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, value Cscalar)

func AtgScatter_ ¶

func AtgScatter_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor)

func AtgSearchsorted ¶ added in v0.3.0

func AtgSearchsorted(ptr *Ctensor, sortedSequence Ctensor, self Ctensor, outInt32 int32, right int32, side string, sorter Ctensor)

func AtgSearchsortedScalar ¶ added in v0.4.0

func AtgSearchsortedScalar(ptr *Ctensor, sortedSequence Ctensor, selfScalar Cscalar, outInt32 int32, right int32, side string, sorter Ctensor)

func AtgSearchsortedScalarOut ¶ added in v0.8.0

func AtgSearchsortedScalarOut(ptr *Ctensor, out Ctensor, sortedSequence Ctensor, selfScalar Cscalar, outInt32 int32, right int32, side string, sorter Ctensor)

func AtgSearchsortedTensorOut ¶ added in v0.4.0

func AtgSearchsortedTensorOut(ptr *Ctensor, out Ctensor, sortedSequence Ctensor, self Ctensor, outInt32 int32, right int32, side string, sorter Ctensor)

func AtgSegmentReduce ¶ added in v0.4.0

func AtgSegmentReduce(ptr *Ctensor, data Ctensor, reduce string, lengths Ctensor, indices Ctensor, offsets Ctensor, axis int64, unsafety int32, initial Cscalar)

func AtgSegmentReduceOut ¶ added in v0.8.0

func AtgSegmentReduceOut(ptr *Ctensor, out Ctensor, data Ctensor, reduce string, lengths Ctensor, indices Ctensor, offsets Ctensor, axis int64, unsafety int32, initial Cscalar)

func AtgSelect ¶

func AtgSelect(ptr *Ctensor, self Ctensor, dim int64, index int64)

func AtgSelectBackward ¶ added in v0.3.0

func AtgSelectBackward(ptr *Ctensor, gradOutput Ctensor, inputSizesData []int64, inputSizesLen int, dim int64, index int64)

func AtgSelectBackwardOut ¶ added in v0.8.0

func AtgSelectBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, inputSizesData []int64, inputSizesLen int, dim int64, index int64)

func AtgSelectCopy ¶ added in v0.8.0

func AtgSelectCopy(ptr *Ctensor, self Ctensor, dim int64, index int64)

func AtgSelectCopyIntOut ¶ added in v0.8.0

func AtgSelectCopyIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index int64)

func AtgSelectScatter ¶ added in v0.7.0

func AtgSelectScatter(ptr *Ctensor, self Ctensor, src Ctensor, dim int64, index int64)

func AtgSelectScatterOut ¶ added in v0.8.0

func AtgSelectScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, dim int64, index int64)

func AtgSelu ¶

func AtgSelu(ptr *Ctensor, self Ctensor)

func AtgSelu_ ¶

func AtgSelu_(ptr *Ctensor, self Ctensor)

func AtgSet ¶ added in v0.8.0

func AtgSet(ptr *Ctensor, self Ctensor)

func AtgSetOut ¶ added in v0.8.0

func AtgSetOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSetRequiresGrad ¶

func AtgSetRequiresGrad(ptr *Ctensor, self Ctensor, r int32)

func AtgSetSourceTensor ¶ added in v0.8.0

func AtgSetSourceTensor(ptr *Ctensor, self Ctensor, source Ctensor)

func AtgSetSourceTensorOut ¶ added in v0.8.0

func AtgSetSourceTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, source Ctensor)

func AtgSetSourceTensorStorageOffset_ ¶ added in v0.8.0

func AtgSetSourceTensorStorageOffset_(ptr *Ctensor, self Ctensor, source Ctensor, storageOffset int64, sizeData []int64, sizeLen int, strideData []int64, strideLen int)

func AtgSetSourceTensor_ ¶ added in v0.4.0

func AtgSetSourceTensor_(ptr *Ctensor, self Ctensor, source Ctensor)

func AtgSet_ ¶

func AtgSet_(ptr *Ctensor, self Ctensor)

func AtgSgn ¶ added in v0.3.0

func AtgSgn(ptr *Ctensor, self Ctensor)

func AtgSgnOut ¶ added in v0.3.0

func AtgSgnOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSgn_ ¶ added in v0.3.0

func AtgSgn_(ptr *Ctensor, self Ctensor)

func AtgSigmoid ¶

func AtgSigmoid(ptr *Ctensor, self Ctensor)

func AtgSigmoidBackward ¶

func AtgSigmoidBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor)

func AtgSigmoidBackwardGradInput ¶ added in v0.4.0

func AtgSigmoidBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, output Ctensor)

func AtgSigmoidOut ¶

func AtgSigmoidOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSigmoid_ ¶

func AtgSigmoid_(ptr *Ctensor, self Ctensor)

func AtgSign ¶

func AtgSign(ptr *Ctensor, self Ctensor)

func AtgSignOut ¶

func AtgSignOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSign_ ¶

func AtgSign_(ptr *Ctensor, self Ctensor)

func AtgSignbit ¶ added in v0.3.0

func AtgSignbit(ptr *Ctensor, self Ctensor)

func AtgSignbitOut ¶ added in v0.3.0

func AtgSignbitOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSilu ¶ added in v0.3.0

func AtgSilu(ptr *Ctensor, self Ctensor)

func AtgSiluBackward ¶ added in v0.3.0

func AtgSiluBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)

func AtgSiluBackwardGradInput ¶ added in v0.5.0

func AtgSiluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor)

func AtgSiluOut ¶ added in v0.3.0

func AtgSiluOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSilu_ ¶ added in v0.3.0

func AtgSilu_(ptr *Ctensor, self Ctensor)

func AtgSin ¶

func AtgSin(ptr *Ctensor, self Ctensor)

func AtgSinOut ¶

func AtgSinOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSin_ ¶

func AtgSin_(ptr *Ctensor, self Ctensor)

func AtgSinc ¶ added in v0.4.0

func AtgSinc(ptr *Ctensor, self Ctensor)

func AtgSincOut ¶ added in v0.4.0

func AtgSincOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSinc_ ¶ added in v0.4.0

func AtgSinc_(ptr *Ctensor, self Ctensor)

func AtgSinh ¶

func AtgSinh(ptr *Ctensor, self Ctensor)

func AtgSinhOut ¶

func AtgSinhOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSinh_ ¶

func AtgSinh_(ptr *Ctensor, self Ctensor)

func AtgSlice ¶

func AtgSlice(ptr *Ctensor, self Ctensor, dim int64, startVal int64, startNull int, endVal int64, endNull int, step int64)

func AtgSliceBackward ¶ added in v0.3.0

func AtgSliceBackward(ptr *Ctensor, gradOutput Ctensor, inputSizesData []int64, inputSizesLen int, dim int64, start int64, end int64, step int64)

func AtgSliceBackwardOut ¶ added in v0.8.0

func AtgSliceBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, inputSizesData []int64, inputSizesLen int, dim int64, start int64, end int64, step int64)

func AtgSliceCopy ¶ added in v0.8.0

func AtgSliceCopy(ptr *Ctensor, self Ctensor, dim int64, startVal int64, startNull int, endVal int64, endNull int, step int64)

func AtgSliceCopyTensorOut ¶ added in v0.8.0

func AtgSliceCopyTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, startVal int64, startNull int, endVal int64, endNull int, step int64)

func AtgSliceScatter ¶ added in v0.7.0

func AtgSliceScatter(ptr *Ctensor, self Ctensor, src Ctensor, dim int64, startVal int64, startNull int, endVal int64, endNull int, step int64)

func AtgSliceScatterOut ¶ added in v0.8.0

func AtgSliceScatterOut(ptr *Ctensor, out Ctensor, self Ctensor, src Ctensor, dim int64, startVal int64, startNull int, endVal int64, endNull int, step int64)

func AtgSlogdet ¶

func AtgSlogdet(ptr *Ctensor, self Ctensor)

func AtgSlogdetOut ¶ added in v0.8.0

func AtgSlogdetOut(ptr *Ctensor, sign Ctensor, logabsdet Ctensor, self Ctensor)

func AtgSlowConv3d ¶

func AtgSlowConv3d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int)

func AtgSlowConv3dOut ¶

func AtgSlowConv3dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int)

func AtgSlowConvDilated2d ¶

func AtgSlowConvDilated2d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int)

func AtgSlowConvDilated2dOut ¶ added in v0.8.0

func AtgSlowConvDilated2dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int)

func AtgSlowConvDilated3d ¶

func AtgSlowConvDilated3d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int)

func AtgSlowConvDilated3dOut ¶ added in v0.8.0

func AtgSlowConvDilated3dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int)

func AtgSlowConvTranspose2d ¶

func AtgSlowConvTranspose2d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, dilationData []int64, dilationLen int)

func AtgSlowConvTranspose2dOut ¶

func AtgSlowConvTranspose2dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, dilationData []int64, dilationLen int)

func AtgSlowConvTranspose3d ¶

func AtgSlowConvTranspose3d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, dilationData []int64, dilationLen int)

func AtgSlowConvTranspose3dOut ¶

func AtgSlowConvTranspose3dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, dilationData []int64, dilationLen int)

func AtgSmm ¶

func AtgSmm(ptr *Ctensor, self Ctensor, mat2 Ctensor)

func AtgSmoothL1Loss ¶

func AtgSmoothL1Loss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64, beta float64)

func AtgSmoothL1LossBackward ¶

func AtgSmoothL1LossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64, beta float64)

func AtgSmoothL1LossBackwardGradInput ¶ added in v0.4.0

func AtgSmoothL1LossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64, beta float64)

func AtgSmoothL1LossOut ¶

func AtgSmoothL1LossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64, beta float64)

func AtgSoftMarginLoss ¶

func AtgSoftMarginLoss(ptr *Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgSoftMarginLossBackward ¶

func AtgSoftMarginLossBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgSoftMarginLossBackwardGradInput ¶ added in v0.4.0

func AtgSoftMarginLossBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgSoftMarginLossOut ¶

func AtgSoftMarginLossOut(ptr *Ctensor, out Ctensor, self Ctensor, target Ctensor, reduction int64)

func AtgSoftmax ¶

func AtgSoftmax(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func AtgSoftmaxIntOut ¶ added in v0.8.0

func AtgSoftmaxIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, dtype int32)

func AtgSoftplus ¶

func AtgSoftplus(ptr *Ctensor, self Ctensor)

func AtgSoftplusBackward ¶

func AtgSoftplusBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, beta Cscalar, threshold Cscalar)

func AtgSoftplusBackwardGradInput ¶ added in v0.4.0

func AtgSoftplusBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, beta Cscalar, threshold Cscalar)

func AtgSoftplusOut ¶

func AtgSoftplusOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSoftshrink ¶

func AtgSoftshrink(ptr *Ctensor, self Ctensor)

func AtgSoftshrinkBackward ¶

func AtgSoftshrinkBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, lambd Cscalar)

func AtgSoftshrinkBackwardGradInput ¶ added in v0.4.0

func AtgSoftshrinkBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, lambd Cscalar)

func AtgSoftshrinkOut ¶

func AtgSoftshrinkOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSort ¶

func AtgSort(ptr *Ctensor, self Ctensor, dim int64, descending int32)

func AtgSortStable ¶ added in v0.4.0

func AtgSortStable(ptr *Ctensor, self Ctensor, stable int32, dim int64, descending int32)

func AtgSortValues ¶ added in v0.4.0

func AtgSortValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, dim int64, descending int32)

func AtgSortValuesStable ¶ added in v0.4.0

func AtgSortValuesStable(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, stable int32, dim int64, descending int32)

func AtgSparseBscTensor ¶ added in v0.8.0

func AtgSparseBscTensor(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, optionsKind int32, optionsDevice int32)

func AtgSparseBscTensorCcolRowValueSize ¶ added in v0.8.0

func AtgSparseBscTensorCcolRowValueSize(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgSparseBsrTensor ¶ added in v0.8.0

func AtgSparseBsrTensor(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, optionsKind int32, optionsDevice int32)

func AtgSparseBsrTensorCrowColValueSize ¶ added in v0.8.0

func AtgSparseBsrTensorCrowColValueSize(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgSparseCompressedTensor ¶ added in v0.8.0

func AtgSparseCompressedTensor(ptr *Ctensor, compressedIndices Ctensor, plainIndices Ctensor, values Ctensor, optionsKind int32, optionsDevice int32)

func AtgSparseCompressedTensorCompPlainValueSize ¶ added in v0.8.0

func AtgSparseCompressedTensorCompPlainValueSize(ptr *Ctensor, compressedIndices Ctensor, plainIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgSparseCooTensor ¶

func AtgSparseCooTensor(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgSparseCooTensorIndices ¶ added in v0.4.0

func AtgSparseCooTensorIndices(ptr *Ctensor, indices Ctensor, values Ctensor, optionsKind int32, optionsDevice int32, isCoalesced int32)

func AtgSparseCooTensorIndicesSize ¶ added in v0.4.0

func AtgSparseCooTensorIndicesSize(ptr *Ctensor, indices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32, isCoalesced int32)

func AtgSparseCooTensorSizeOut ¶ added in v0.8.0

func AtgSparseCooTensorSizeOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)

func AtgSparseCscTensor ¶ added in v0.8.0

func AtgSparseCscTensor(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, optionsKind int32, optionsDevice int32)

func AtgSparseCscTensorCcolRowValueSize ¶ added in v0.8.0

func AtgSparseCscTensorCcolRowValueSize(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgSparseCsrTensor ¶ added in v0.5.0

func AtgSparseCsrTensor(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, optionsKind int32, optionsDevice int32)

func AtgSparseCsrTensorCrowColValueSize ¶ added in v0.5.0

func AtgSparseCsrTensorCrowColValueSize(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgSparseDim ¶ added in v0.4.0

func AtgSparseDim(self Ctensor) int64

func AtgSparseMask ¶

func AtgSparseMask(ptr *Ctensor, self Ctensor, mask Ctensor)

func AtgSparseMaskOut ¶ added in v0.8.0

func AtgSparseMaskOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor)

func AtgSparseResize ¶ added in v0.8.0

func AtgSparseResize(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, denseDim int64)

func AtgSparseResizeAndClear ¶ added in v0.8.0

func AtgSparseResizeAndClear(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, denseDim int64)

func AtgSparseResizeAndClearOut ¶ added in v0.8.0

func AtgSparseResizeAndClearOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, denseDim int64)

func AtgSparseResizeAndClear_ ¶

func AtgSparseResizeAndClear_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, denseDim int64)

func AtgSparseResizeOut ¶ added in v0.8.0

func AtgSparseResizeOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, denseDim int64)

func AtgSparseResize_ ¶

func AtgSparseResize_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, sparseDim int64, denseDim int64)

func AtgSparseSampledAddmm ¶ added in v0.7.0

func AtgSparseSampledAddmm(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)

func AtgSparseSampledAddmmOut ¶ added in v0.7.0

func AtgSparseSampledAddmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)

func AtgSpecialAiryAi ¶ added in v0.8.0

func AtgSpecialAiryAi(ptr *Ctensor, x Ctensor)

func AtgSpecialAiryAiOut ¶ added in v0.8.0

func AtgSpecialAiryAiOut(ptr *Ctensor, out Ctensor, x Ctensor)

func AtgSpecialBesselJ0 ¶ added in v0.8.0

func AtgSpecialBesselJ0(ptr *Ctensor, self Ctensor)

func AtgSpecialBesselJ0Out ¶ added in v0.8.0

func AtgSpecialBesselJ0Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialBesselJ1 ¶ added in v0.8.0

func AtgSpecialBesselJ1(ptr *Ctensor, self Ctensor)

func AtgSpecialBesselJ1Out ¶ added in v0.8.0

func AtgSpecialBesselJ1Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialBesselY0 ¶ added in v0.8.0

func AtgSpecialBesselY0(ptr *Ctensor, self Ctensor)

func AtgSpecialBesselY0Out ¶ added in v0.8.0

func AtgSpecialBesselY0Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialBesselY1 ¶ added in v0.8.0

func AtgSpecialBesselY1(ptr *Ctensor, self Ctensor)

func AtgSpecialBesselY1Out ¶ added in v0.8.0

func AtgSpecialBesselY1Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialChebyshevPolynomialT ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialT(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialChebyshevPolynomialTNScalar ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialTNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialChebyshevPolynomialTNScalarOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialTNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialChebyshevPolynomialTOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialTOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialChebyshevPolynomialTXScalar ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialTXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialChebyshevPolynomialTXScalarOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialTXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialChebyshevPolynomialU ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialU(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialChebyshevPolynomialUNScalar ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialUNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialChebyshevPolynomialUNScalarOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialUNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialChebyshevPolynomialUOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialUOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialChebyshevPolynomialUXScalar ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialUXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialChebyshevPolynomialUXScalarOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialUXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialChebyshevPolynomialV ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialV(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialChebyshevPolynomialVNScalar ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialVNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialChebyshevPolynomialVNScalarOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialVNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialChebyshevPolynomialVOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialVOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialChebyshevPolynomialVXScalar ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialVXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialChebyshevPolynomialVXScalarOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialVXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialChebyshevPolynomialW ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialW(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialChebyshevPolynomialWNScalar ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialWNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialChebyshevPolynomialWNScalarOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialWNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialChebyshevPolynomialWOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialWOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialChebyshevPolynomialWXScalar ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialWXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialChebyshevPolynomialWXScalarOut ¶ added in v0.8.0

func AtgSpecialChebyshevPolynomialWXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialDigamma ¶ added in v0.5.0

func AtgSpecialDigamma(ptr *Ctensor, self Ctensor)

func AtgSpecialDigammaOut ¶ added in v0.5.0

func AtgSpecialDigammaOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialEntr ¶ added in v0.4.0

func AtgSpecialEntr(ptr *Ctensor, self Ctensor)

func AtgSpecialEntrOut ¶ added in v0.4.0

func AtgSpecialEntrOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialErf ¶ added in v0.4.0

func AtgSpecialErf(ptr *Ctensor, self Ctensor)

func AtgSpecialErfOut ¶ added in v0.4.0

func AtgSpecialErfOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialErfc ¶ added in v0.4.0

func AtgSpecialErfc(ptr *Ctensor, self Ctensor)

func AtgSpecialErfcOut ¶ added in v0.4.0

func AtgSpecialErfcOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialErfcx ¶ added in v0.5.0

func AtgSpecialErfcx(ptr *Ctensor, self Ctensor)

func AtgSpecialErfcxOut ¶ added in v0.5.0

func AtgSpecialErfcxOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialErfinv ¶ added in v0.4.0

func AtgSpecialErfinv(ptr *Ctensor, self Ctensor)

func AtgSpecialErfinvOut ¶ added in v0.4.0

func AtgSpecialErfinvOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialExp2 ¶ added in v0.4.0

func AtgSpecialExp2(ptr *Ctensor, self Ctensor)

func AtgSpecialExp2Out ¶ added in v0.4.0

func AtgSpecialExp2Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialExpit ¶ added in v0.4.0

func AtgSpecialExpit(ptr *Ctensor, self Ctensor)

func AtgSpecialExpitOut ¶ added in v0.4.0

func AtgSpecialExpitOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialExpm1 ¶ added in v0.4.0

func AtgSpecialExpm1(ptr *Ctensor, self Ctensor)

func AtgSpecialExpm1Out ¶ added in v0.4.0

func AtgSpecialExpm1Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialGammainc ¶ added in v0.5.0

func AtgSpecialGammainc(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgSpecialGammaincOut ¶ added in v0.5.0

func AtgSpecialGammaincOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgSpecialGammaincc ¶ added in v0.5.0

func AtgSpecialGammaincc(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgSpecialGammainccOut ¶ added in v0.5.0

func AtgSpecialGammainccOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgSpecialGammaln ¶ added in v0.4.0

func AtgSpecialGammaln(ptr *Ctensor, self Ctensor)

func AtgSpecialGammalnOut ¶ added in v0.4.0

func AtgSpecialGammalnOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialHermitePolynomialH ¶ added in v0.8.0

func AtgSpecialHermitePolynomialH(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialHermitePolynomialHNScalar ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialHermitePolynomialHNScalarOut ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialHermitePolynomialHOut ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialHermitePolynomialHXScalar ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialHermitePolynomialHXScalarOut ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialHermitePolynomialHe ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHe(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialHermitePolynomialHeNScalar ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHeNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialHermitePolynomialHeNScalarOut ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHeNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialHermitePolynomialHeOut ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHeOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialHermitePolynomialHeXScalar ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHeXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialHermitePolynomialHeXScalarOut ¶ added in v0.8.0

func AtgSpecialHermitePolynomialHeXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialI0 ¶ added in v0.5.0

func AtgSpecialI0(ptr *Ctensor, self Ctensor)

func AtgSpecialI0Out ¶ added in v0.5.0

func AtgSpecialI0Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialI0e ¶ added in v0.4.0

func AtgSpecialI0e(ptr *Ctensor, self Ctensor)

func AtgSpecialI0eOut ¶ added in v0.4.0

func AtgSpecialI0eOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialI1 ¶ added in v0.5.0

func AtgSpecialI1(ptr *Ctensor, self Ctensor)

func AtgSpecialI1Out ¶ added in v0.5.0

func AtgSpecialI1Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialI1e ¶ added in v0.5.0

func AtgSpecialI1e(ptr *Ctensor, self Ctensor)

func AtgSpecialI1eOut ¶ added in v0.5.0

func AtgSpecialI1eOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialLaguerrePolynomialL ¶ added in v0.8.0

func AtgSpecialLaguerrePolynomialL(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialLaguerrePolynomialLNScalar ¶ added in v0.8.0

func AtgSpecialLaguerrePolynomialLNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialLaguerrePolynomialLNScalarOut ¶ added in v0.8.0

func AtgSpecialLaguerrePolynomialLNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialLaguerrePolynomialLOut ¶ added in v0.8.0

func AtgSpecialLaguerrePolynomialLOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialLaguerrePolynomialLXScalar ¶ added in v0.8.0

func AtgSpecialLaguerrePolynomialLXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialLaguerrePolynomialLXScalarOut ¶ added in v0.8.0

func AtgSpecialLaguerrePolynomialLXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialLegendrePolynomialP ¶ added in v0.8.0

func AtgSpecialLegendrePolynomialP(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialLegendrePolynomialPNScalar ¶ added in v0.8.0

func AtgSpecialLegendrePolynomialPNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialLegendrePolynomialPNScalarOut ¶ added in v0.8.0

func AtgSpecialLegendrePolynomialPNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialLegendrePolynomialPOut ¶ added in v0.8.0

func AtgSpecialLegendrePolynomialPOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialLegendrePolynomialPXScalar ¶ added in v0.8.0

func AtgSpecialLegendrePolynomialPXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialLegendrePolynomialPXScalarOut ¶ added in v0.8.0

func AtgSpecialLegendrePolynomialPXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialLog1p ¶ added in v0.5.0

func AtgSpecialLog1p(ptr *Ctensor, self Ctensor)

func AtgSpecialLog1pOut ¶ added in v0.5.0

func AtgSpecialLog1pOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialLogNdtr ¶ added in v0.8.0

func AtgSpecialLogNdtr(ptr *Ctensor, self Ctensor)

func AtgSpecialLogNdtrOut ¶ added in v0.8.0

func AtgSpecialLogNdtrOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialLogSoftmax ¶ added in v0.5.0

func AtgSpecialLogSoftmax(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func AtgSpecialLogit ¶ added in v0.4.0

func AtgSpecialLogit(ptr *Ctensor, self Ctensor, epsVal float64, epsNull int)

func AtgSpecialLogitOut ¶ added in v0.4.0

func AtgSpecialLogitOut(ptr *Ctensor, out Ctensor, self Ctensor, epsVal float64, epsNull int)

func AtgSpecialLogsumexp ¶ added in v0.5.0

func AtgSpecialLogsumexp(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgSpecialLogsumexpOut ¶ added in v0.5.0

func AtgSpecialLogsumexpOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func AtgSpecialModifiedBesselI0 ¶ added in v0.8.0

func AtgSpecialModifiedBesselI0(ptr *Ctensor, self Ctensor)

func AtgSpecialModifiedBesselI0Out ¶ added in v0.8.0

func AtgSpecialModifiedBesselI0Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialModifiedBesselI1 ¶ added in v0.8.0

func AtgSpecialModifiedBesselI1(ptr *Ctensor, self Ctensor)

func AtgSpecialModifiedBesselI1Out ¶ added in v0.8.0

func AtgSpecialModifiedBesselI1Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialModifiedBesselK0 ¶ added in v0.8.0

func AtgSpecialModifiedBesselK0(ptr *Ctensor, self Ctensor)

func AtgSpecialModifiedBesselK0Out ¶ added in v0.8.0

func AtgSpecialModifiedBesselK0Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialModifiedBesselK1 ¶ added in v0.8.0

func AtgSpecialModifiedBesselK1(ptr *Ctensor, self Ctensor)

func AtgSpecialModifiedBesselK1Out ¶ added in v0.8.0

func AtgSpecialModifiedBesselK1Out(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialMultigammaln ¶ added in v0.5.0

func AtgSpecialMultigammaln(ptr *Ctensor, self Ctensor, p int64)

func AtgSpecialMultigammalnOut ¶ added in v0.5.0

func AtgSpecialMultigammalnOut(ptr *Ctensor, out Ctensor, self Ctensor, p int64)

func AtgSpecialNdtr ¶ added in v0.5.0

func AtgSpecialNdtr(ptr *Ctensor, self Ctensor)

func AtgSpecialNdtrOut ¶ added in v0.5.0

func AtgSpecialNdtrOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialNdtri ¶ added in v0.5.0

func AtgSpecialNdtri(ptr *Ctensor, self Ctensor)

func AtgSpecialNdtriOut ¶ added in v0.5.0

func AtgSpecialNdtriOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialPolygamma ¶ added in v0.5.0

func AtgSpecialPolygamma(ptr *Ctensor, n int64, self Ctensor)

func AtgSpecialPolygammaOut ¶ added in v0.5.0

func AtgSpecialPolygammaOut(ptr *Ctensor, out Ctensor, n int64, self Ctensor)

func AtgSpecialPsi ¶ added in v0.5.0

func AtgSpecialPsi(ptr *Ctensor, self Ctensor)

func AtgSpecialPsiOut ¶ added in v0.5.0

func AtgSpecialPsiOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialRound ¶ added in v0.5.0

func AtgSpecialRound(ptr *Ctensor, self Ctensor, decimals int64)

func AtgSpecialRoundOut ¶ added in v0.5.0

func AtgSpecialRoundOut(ptr *Ctensor, out Ctensor, self Ctensor, decimals int64)

func AtgSpecialScaledModifiedBesselK0 ¶ added in v0.8.0

func AtgSpecialScaledModifiedBesselK0(ptr *Ctensor, x Ctensor)

func AtgSpecialScaledModifiedBesselK0Out ¶ added in v0.8.0

func AtgSpecialScaledModifiedBesselK0Out(ptr *Ctensor, out Ctensor, x Ctensor)

func AtgSpecialScaledModifiedBesselK1 ¶ added in v0.8.0

func AtgSpecialScaledModifiedBesselK1(ptr *Ctensor, x Ctensor)

func AtgSpecialScaledModifiedBesselK1Out ¶ added in v0.8.0

func AtgSpecialScaledModifiedBesselK1Out(ptr *Ctensor, out Ctensor, x Ctensor)

func AtgSpecialShiftedChebyshevPolynomialT ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialT(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialTNScalar ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialTNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialShiftedChebyshevPolynomialTNScalarOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialTNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialShiftedChebyshevPolynomialTOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialTOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialTXScalar ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialTXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialTXScalarOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialTXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialU ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialU(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialUNScalar ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialUNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialShiftedChebyshevPolynomialUNScalarOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialUNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialShiftedChebyshevPolynomialUOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialUOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialUXScalar ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialUXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialUXScalarOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialUXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialV ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialV(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialVNScalar ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialVNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialShiftedChebyshevPolynomialVNScalarOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialVNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialShiftedChebyshevPolynomialVOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialVOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialVXScalar ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialVXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialVXScalarOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialVXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialW ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialW(ptr *Ctensor, x Ctensor, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialWNScalar ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialWNScalar(ptr *Ctensor, x Ctensor, n Cscalar)

func AtgSpecialShiftedChebyshevPolynomialWNScalarOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialWNScalarOut(ptr *Ctensor, out Ctensor, x Ctensor, n Cscalar)

func AtgSpecialShiftedChebyshevPolynomialWOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialWOut(ptr *Ctensor, out Ctensor, x Ctensor, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialWXScalar ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialWXScalar(ptr *Ctensor, x Cscalar, n Ctensor)

func AtgSpecialShiftedChebyshevPolynomialWXScalarOut ¶ added in v0.8.0

func AtgSpecialShiftedChebyshevPolynomialWXScalarOut(ptr *Ctensor, out Ctensor, x Cscalar, n Ctensor)

func AtgSpecialSinc ¶ added in v0.5.0

func AtgSpecialSinc(ptr *Ctensor, self Ctensor)

func AtgSpecialSincOut ¶ added in v0.5.0

func AtgSpecialSincOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSpecialSoftmax ¶ added in v0.7.0

func AtgSpecialSoftmax(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func AtgSpecialSphericalBesselJ0 ¶ added in v0.8.0

func AtgSpecialSphericalBesselJ0(ptr *Ctensor, x Ctensor)

func AtgSpecialSphericalBesselJ0Out ¶ added in v0.8.0

func AtgSpecialSphericalBesselJ0Out(ptr *Ctensor, out Ctensor, x Ctensor)

func AtgSpecialXlog1py ¶ added in v0.4.0

func AtgSpecialXlog1py(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgSpecialXlog1pyOtherScalar ¶ added in v0.4.0

func AtgSpecialXlog1pyOtherScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgSpecialXlog1pyOtherScalarOut ¶ added in v0.4.0

func AtgSpecialXlog1pyOtherScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgSpecialXlog1pyOut ¶ added in v0.4.0

func AtgSpecialXlog1pyOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgSpecialXlog1pySelfScalar ¶ added in v0.4.0

func AtgSpecialXlog1pySelfScalar(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgSpecialXlog1pySelfScalarOut ¶ added in v0.4.0

func AtgSpecialXlog1pySelfScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgSpecialXlogy ¶ added in v0.5.0

func AtgSpecialXlogy(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgSpecialXlogyOtherScalar ¶ added in v0.5.0

func AtgSpecialXlogyOtherScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgSpecialXlogyOtherScalarOut ¶ added in v0.5.0

func AtgSpecialXlogyOtherScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgSpecialXlogyOut ¶ added in v0.5.0

func AtgSpecialXlogyOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgSpecialXlogySelfScalar ¶ added in v0.5.0

func AtgSpecialXlogySelfScalar(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgSpecialXlogySelfScalarOut ¶ added in v0.5.0

func AtgSpecialXlogySelfScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgSpecialZeta ¶ added in v0.5.0

func AtgSpecialZeta(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgSpecialZetaOtherScalar ¶ added in v0.5.0

func AtgSpecialZetaOtherScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgSpecialZetaOtherScalarOut ¶ added in v0.5.0

func AtgSpecialZetaOtherScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgSpecialZetaOut ¶ added in v0.5.0

func AtgSpecialZetaOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgSpecialZetaSelfScalar ¶ added in v0.5.0

func AtgSpecialZetaSelfScalar(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgSpecialZetaSelfScalarOut ¶ added in v0.5.0

func AtgSpecialZetaSelfScalarOut(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgSqrt ¶

func AtgSqrt(ptr *Ctensor, self Ctensor)

func AtgSqrtOut ¶

func AtgSqrtOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSqrt_ ¶

func AtgSqrt_(ptr *Ctensor, self Ctensor)

func AtgSquare ¶

func AtgSquare(ptr *Ctensor, self Ctensor)

func AtgSquareOut ¶ added in v0.4.0

func AtgSquareOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSquare_ ¶

func AtgSquare_(ptr *Ctensor, self Ctensor)

func AtgSqueeze ¶

func AtgSqueeze(ptr *Ctensor, self Ctensor)

func AtgSqueezeCopy ¶ added in v0.8.0

func AtgSqueezeCopy(ptr *Ctensor, self Ctensor)

func AtgSqueezeCopyDim ¶ added in v0.8.0

func AtgSqueezeCopyDim(ptr *Ctensor, self Ctensor, dim int64)

func AtgSqueezeCopyDimOut ¶ added in v0.8.0

func AtgSqueezeCopyDimOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)

func AtgSqueezeCopyDims ¶ added in v0.8.0

func AtgSqueezeCopyDims(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)

func AtgSqueezeCopyDimsOut ¶ added in v0.8.0

func AtgSqueezeCopyDimsOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int)

func AtgSqueezeCopyOut ¶ added in v0.8.0

func AtgSqueezeCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgSqueezeDim ¶ added in v0.4.0

func AtgSqueezeDim(ptr *Ctensor, self Ctensor, dim int64)

func AtgSqueezeDim_ ¶ added in v0.4.0

func AtgSqueezeDim_(ptr *Ctensor, self Ctensor, dim int64)

func AtgSqueezeDims ¶ added in v0.8.0

func AtgSqueezeDims(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)

func AtgSqueezeDims_ ¶ added in v0.8.0

func AtgSqueezeDims_(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)

func AtgSqueeze_ ¶

func AtgSqueeze_(ptr *Ctensor, self Ctensor)

func AtgSspaddmm ¶

func AtgSspaddmm(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)

func AtgSspaddmmOut ¶

func AtgSspaddmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)

func AtgStack ¶

func AtgStack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func AtgStackOut ¶

func AtgStackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func AtgStd ¶

func AtgStd(ptr *Ctensor, self Ctensor, unbiased int32)

func AtgStdCorrection ¶ added in v0.4.0

func AtgStdCorrection(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, keepdim int32)

func AtgStdCorrectionOut ¶ added in v0.4.0

func AtgStdCorrectionOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, keepdim int32)

func AtgStdDim ¶ added in v0.4.0

func AtgStdDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, keepdim int32)

func AtgStdMean ¶

func AtgStdMean(ptr *Ctensor, self Ctensor, unbiased int32)

func AtgStdMeanCorrection ¶ added in v0.4.0

func AtgStdMeanCorrection(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, keepdim int32)

func AtgStdMeanCorrectionOut ¶ added in v0.8.0

func AtgStdMeanCorrectionOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, keepdim int32)

func AtgStdMeanDim ¶ added in v0.4.0

func AtgStdMeanDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, keepdim int32)

func AtgStdOut ¶

func AtgStdOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, keepdim int32)

func AtgStft ¶

func AtgStft(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, winLengthVal int64, winLengthNull int, window Ctensor, normalized int32, onesided int32, returnComplex int32)

func AtgStftCenter ¶ added in v0.8.0

func AtgStftCenter(ptr *Ctensor, self Ctensor, nFft int64, hopLengthVal int64, hopLengthNull int, winLengthVal int64, winLengthNull int, window Ctensor, center int32, padMode string, normalized int32, onesided int32, returnComplex int32)

func AtgSub ¶

func AtgSub(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgSubOut ¶

func AtgSubOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgSubScalar ¶ added in v0.4.0

func AtgSubScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgSubScalarOut ¶ added in v0.8.0

func AtgSubScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgSubScalar_ ¶ added in v0.4.0

func AtgSubScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgSub_ ¶

func AtgSub_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgSubtract ¶ added in v0.3.0

func AtgSubtract(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgSubtractOut ¶ added in v0.3.0

func AtgSubtractOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgSubtractScalar ¶ added in v0.4.0

func AtgSubtractScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgSubtractScalar_ ¶ added in v0.4.0

func AtgSubtractScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgSubtract_ ¶ added in v0.3.0

func AtgSubtract_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgSum ¶

func AtgSum(ptr *Ctensor, self Ctensor, dtype int32)

func AtgSumDimIntlist ¶ added in v0.4.0

func AtgSumDimIntlist(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgSumIntlistOut ¶ added in v0.4.0

func AtgSumIntlistOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func AtgSumOut ¶

func AtgSumOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32)

func AtgSumToSize ¶

func AtgSumToSize(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgSvd ¶

func AtgSvd(ptr *Ctensor, self Ctensor, some int32, computeUv int32)

func AtgSvdU ¶ added in v0.4.0

func AtgSvdU(ptr *Ctensor, u Ctensor, s Ctensor, v Ctensor, self Ctensor, some int32, computeUv int32)

func AtgSwapaxes ¶ added in v0.4.0

func AtgSwapaxes(ptr *Ctensor, self Ctensor, axis0 int64, axis1 int64)

func AtgSwapaxes_ ¶ added in v0.4.0

func AtgSwapaxes_(ptr *Ctensor, self Ctensor, axis0 int64, axis1 int64)

func AtgSwapdims ¶ added in v0.4.0

func AtgSwapdims(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)

func AtgSwapdims_ ¶ added in v0.4.0

func AtgSwapdims_(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)

func AtgT ¶

func AtgT(ptr *Ctensor, self Ctensor)

func AtgTCopy ¶ added in v0.8.0

func AtgTCopy(ptr *Ctensor, self Ctensor)

func AtgTCopyOut ¶ added in v0.8.0

func AtgTCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgT_ ¶

func AtgT_(ptr *Ctensor, self Ctensor)

func AtgTake ¶

func AtgTake(ptr *Ctensor, self Ctensor, index Ctensor)

func AtgTakeAlongDim ¶ added in v0.4.0

func AtgTakeAlongDim(ptr *Ctensor, self Ctensor, indices Ctensor, dimVal int64, dimNull int)

func AtgTakeAlongDimOut ¶ added in v0.4.0

func AtgTakeAlongDimOut(ptr *Ctensor, out Ctensor, self Ctensor, indices Ctensor, dimVal int64, dimNull int)

func AtgTakeOut ¶

func AtgTakeOut(ptr *Ctensor, out Ctensor, self Ctensor, index Ctensor)

func AtgTan ¶

func AtgTan(ptr *Ctensor, self Ctensor)

func AtgTanOut ¶

func AtgTanOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgTan_ ¶

func AtgTan_(ptr *Ctensor, self Ctensor)

func AtgTanh ¶

func AtgTanh(ptr *Ctensor, self Ctensor)

func AtgTanhBackward ¶

func AtgTanhBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor)

func AtgTanhBackwardGradInput ¶ added in v0.4.0

func AtgTanhBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, output Ctensor)

func AtgTanhOut ¶

func AtgTanhOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgTanh_ ¶

func AtgTanh_(ptr *Ctensor, self Ctensor)

func AtgTensordot ¶

func AtgTensordot(ptr *Ctensor, self Ctensor, other Ctensor, dimsSelfData []int64, dimsSelfLen int, dimsOtherData []int64, dimsOtherLen int)

func AtgTensordotOut ¶ added in v0.4.0

func AtgTensordotOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, dimsSelfData []int64, dimsSelfLen int, dimsOtherData []int64, dimsOtherLen int)

func AtgThreshold ¶

func AtgThreshold(ptr *Ctensor, self Ctensor, threshold Cscalar, value Cscalar)

func AtgThresholdBackward ¶

func AtgThresholdBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, threshold Cscalar)

func AtgThresholdBackwardGradInput ¶ added in v0.4.0

func AtgThresholdBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, threshold Cscalar)

func AtgThresholdOut ¶

func AtgThresholdOut(ptr *Ctensor, out Ctensor, self Ctensor, threshold Cscalar, value Cscalar)

func AtgThreshold_ ¶

func AtgThreshold_(ptr *Ctensor, self Ctensor, threshold Cscalar, value Cscalar)

func AtgTile ¶ added in v0.4.0

func AtgTile(ptr *Ctensor, self Ctensor, dimsData []int64, dimsLen int)

func AtgTo ¶

func AtgTo(ptr *Ctensor, self Ctensor, device int32)

func AtgToDense ¶

func AtgToDense(ptr *Ctensor, self Ctensor, dtype int32, maskedGrad int32)

func AtgToDenseBackward ¶

func AtgToDenseBackward(ptr *Ctensor, grad Ctensor, input Ctensor, maskedGrad int32)

func AtgToDevice ¶ added in v0.4.0

func AtgToDevice(ptr *Ctensor, self Ctensor, device int32, dtype int32, nonBlocking int32, copy int32)

func AtgToDtype ¶ added in v0.4.0

func AtgToDtype(ptr *Ctensor, self Ctensor, dtype int32, nonBlocking int32, copy int32)

func AtgToDtypeLayout ¶ added in v0.4.0

func AtgToDtypeLayout(ptr *Ctensor, self Ctensor, optionsKind int32, optionsDevice int32, nonBlocking int32, copy int32)

func AtgToMkldnn ¶

func AtgToMkldnn(ptr *Ctensor, self Ctensor, dtype int32)

func AtgToMkldnnBackward ¶

func AtgToMkldnnBackward(ptr *Ctensor, grad Ctensor, input Ctensor)

func AtgToMkldnnOut ¶ added in v0.8.0

func AtgToMkldnnOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32)

func AtgToOther ¶ added in v0.4.0

func AtgToOther(ptr *Ctensor, self Ctensor, other Ctensor, nonBlocking int32, copy int32)

func AtgToPaddedTensor ¶ added in v0.8.0

func AtgToPaddedTensor(ptr *Ctensor, self Ctensor, padding float64, outputSizeData []int64, outputSizeLen int)

func AtgToPaddedTensorOut ¶ added in v0.8.0

func AtgToPaddedTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, padding float64, outputSizeData []int64, outputSizeLen int)

func AtgToSparse ¶

func AtgToSparse(ptr *Ctensor, self Ctensor, layout int8, blocksizeData []int64, blocksizeLen int, denseDimVal int64, denseDimNull int)

func AtgToSparseBsc ¶ added in v0.8.0

func AtgToSparseBsc(ptr *Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, denseDimVal int64, denseDimNull int)

func AtgToSparseBsr ¶ added in v0.8.0

func AtgToSparseBsr(ptr *Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, denseDimVal int64, denseDimNull int)

func AtgToSparseCsc ¶ added in v0.8.0

func AtgToSparseCsc(ptr *Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)

func AtgToSparseCsr ¶ added in v0.8.0

func AtgToSparseCsr(ptr *Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)

func AtgToSparseSparseDim ¶ added in v0.4.0

func AtgToSparseSparseDim(ptr *Ctensor, self Ctensor, sparseDim int64)

func AtgTopk ¶

func AtgTopk(ptr *Ctensor, self Ctensor, k int64, dim int64, largest int32, sorted int32)

func AtgTopkValues ¶ added in v0.4.0

func AtgTopkValues(ptr *Ctensor, values Ctensor, indices Ctensor, self Ctensor, k int64, dim int64, largest int32, sorted int32)

func AtgTotype ¶

func AtgTotype(ptr *Ctensor, self Ctensor, scalarType int32)

func AtgTrace ¶

func AtgTrace(ptr *Ctensor, self Ctensor)

func AtgTraceBackward ¶ added in v0.3.0

func AtgTraceBackward(ptr *Ctensor, grad Ctensor, sizesData []int64, sizesLen int)

func AtgTraceOut ¶ added in v0.8.0

func AtgTraceOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgTranspose ¶

func AtgTranspose(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)

func AtgTransposeCopy ¶ added in v0.8.0

func AtgTransposeCopy(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)

func AtgTransposeCopyIntOut ¶ added in v0.8.0

func AtgTransposeCopyIntOut(ptr *Ctensor, out Ctensor, self Ctensor, dim0 int64, dim1 int64)

func AtgTranspose_ ¶

func AtgTranspose_(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)

func AtgTrapezoid ¶ added in v0.5.0

func AtgTrapezoid(ptr *Ctensor, y Ctensor, dim int64)

func AtgTrapezoidX ¶ added in v0.5.0

func AtgTrapezoidX(ptr *Ctensor, y Ctensor, x Ctensor, dim int64)

func AtgTrapz ¶

func AtgTrapz(ptr *Ctensor, y Ctensor, x Ctensor, dim int64)

func AtgTrapzDx ¶ added in v0.4.0

func AtgTrapzDx(ptr *Ctensor, y Ctensor, dx float64, dim int64)

func AtgTriangularSolve ¶

func AtgTriangularSolve(ptr *Ctensor, self Ctensor, a Ctensor, upper int32, transpose int32, unitriangular int32)

func AtgTriangularSolveX ¶ added in v0.4.0

func AtgTriangularSolveX(ptr *Ctensor, x Ctensor, m Ctensor, self Ctensor, a Ctensor, upper int32, transpose int32, unitriangular int32)

func AtgTril ¶

func AtgTril(ptr *Ctensor, self Ctensor, diagonal int64)

func AtgTrilIndices ¶

func AtgTrilIndices(ptr *Ctensor, row int64, col int64, offset int64, optionsKind int32, optionsDevice int32)

func AtgTrilIndicesOut ¶ added in v0.8.0

func AtgTrilIndicesOut(ptr *Ctensor, out Ctensor, row int64, col int64, offset int64)

func AtgTrilOut ¶

func AtgTrilOut(ptr *Ctensor, out Ctensor, self Ctensor, diagonal int64)

func AtgTril_ ¶

func AtgTril_(ptr *Ctensor, self Ctensor, diagonal int64)

func AtgTripletMarginLoss ¶

func AtgTripletMarginLoss(ptr *Ctensor, anchor Ctensor, positive Ctensor, negative Ctensor, margin float64, p float64, eps float64, swap int32, reduction int64)

func AtgTriu ¶

func AtgTriu(ptr *Ctensor, self Ctensor, diagonal int64)

func AtgTriuIndices ¶

func AtgTriuIndices(ptr *Ctensor, row int64, col int64, offset int64, optionsKind int32, optionsDevice int32)

func AtgTriuIndicesOut ¶ added in v0.8.0

func AtgTriuIndicesOut(ptr *Ctensor, out Ctensor, row int64, col int64, offset int64)

func AtgTriuOut ¶

func AtgTriuOut(ptr *Ctensor, out Ctensor, self Ctensor, diagonal int64)

func AtgTriu_ ¶

func AtgTriu_(ptr *Ctensor, self Ctensor, diagonal int64)

func AtgTrueDivide ¶

func AtgTrueDivide(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgTrueDivideOut ¶

func AtgTrueDivideOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgTrueDivideScalar ¶ added in v0.4.0

func AtgTrueDivideScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgTrueDivideScalar_ ¶ added in v0.4.0

func AtgTrueDivideScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgTrueDivide_ ¶

func AtgTrueDivide_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgTrunc ¶

func AtgTrunc(ptr *Ctensor, self Ctensor)

func AtgTruncOut ¶

func AtgTruncOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgTrunc_ ¶

func AtgTrunc_(ptr *Ctensor, self Ctensor)

func AtgTypeAs ¶

func AtgTypeAs(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgUnflatten ¶ added in v0.3.0

func AtgUnflatten(ptr *Ctensor, self Ctensor, dim int64, sizesData []int64, sizesLen int)

func AtgUnfold ¶

func AtgUnfold(ptr *Ctensor, self Ctensor, dimension int64, size int64, step int64)

func AtgUnfoldBackward ¶ added in v0.3.0

func AtgUnfoldBackward(ptr *Ctensor, gradIn Ctensor, inputSizesData []int64, inputSizesLen int, dim int64, size int64, step int64)

func AtgUnfoldBackwardOut ¶ added in v0.8.0

func AtgUnfoldBackwardOut(ptr *Ctensor, out Ctensor, gradIn Ctensor, inputSizesData []int64, inputSizesLen int, dim int64, size int64, step int64)

func AtgUnfoldCopy ¶ added in v0.8.0

func AtgUnfoldCopy(ptr *Ctensor, self Ctensor, dimension int64, size int64, step int64)

func AtgUnfoldCopyOut ¶ added in v0.8.0

func AtgUnfoldCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dimension int64, size int64, step int64)

func AtgUniform ¶ added in v0.8.0

func AtgUniform(ptr *Ctensor, self Ctensor, from float64, to float64)

func AtgUniformOut ¶ added in v0.8.0

func AtgUniformOut(ptr *Ctensor, out Ctensor, self Ctensor, from float64, to float64)

func AtgUniform_ ¶

func AtgUniform_(ptr *Ctensor, self Ctensor, from float64, to float64)

func AtgUniqueConsecutive ¶

func AtgUniqueConsecutive(ptr *Ctensor, self Ctensor, returnInverse int32, returnCounts int32, dimVal int64, dimNull int)

func AtgUniqueConsecutiveOut ¶ added in v0.8.0

func AtgUniqueConsecutiveOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, self Ctensor, returnInverse int32, returnCounts int32, dimVal int64, dimNull int)

func AtgUniqueDim ¶

func AtgUniqueDim(ptr *Ctensor, self Ctensor, dim int64, sorted int32, returnInverse int32, returnCounts int32)

func AtgUniqueDimConsecutive ¶

func AtgUniqueDimConsecutive(ptr *Ctensor, self Ctensor, dim int64, returnInverse int32, returnCounts int32)

func AtgUniqueDimConsecutiveOut ¶ added in v0.8.0

func AtgUniqueDimConsecutiveOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, self Ctensor, dim int64, returnInverse int32, returnCounts int32)

func AtgUniqueDimOut ¶ added in v0.8.0

func AtgUniqueDimOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, self Ctensor, dim int64, sorted int32, returnInverse int32, returnCounts int32)

func AtgUnsqueeze ¶

func AtgUnsqueeze(ptr *Ctensor, self Ctensor, dim int64)

func AtgUnsqueezeCopy ¶ added in v0.8.0

func AtgUnsqueezeCopy(ptr *Ctensor, self Ctensor, dim int64)

func AtgUnsqueezeCopyOut ¶ added in v0.8.0

func AtgUnsqueezeCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)

func AtgUnsqueeze_ ¶

func AtgUnsqueeze_(ptr *Ctensor, self Ctensor, dim int64)

func AtgUpsampleBicubic2d ¶

func AtgUpsampleBicubic2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleBicubic2dBackward ¶

func AtgUpsampleBicubic2dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleBicubic2dBackwardGradInput ¶ added in v0.4.0

func AtgUpsampleBicubic2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleBicubic2dOut ¶

func AtgUpsampleBicubic2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleBicubic2dVec ¶ added in v0.8.0

func AtgUpsampleBicubic2dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scaleFactorsData []float64, scaleFactorsLen int)

func AtgUpsampleBilinear2d ¶

func AtgUpsampleBilinear2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleBilinear2dBackward ¶

func AtgUpsampleBilinear2dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleBilinear2dBackwardGradInput ¶ added in v0.4.0

func AtgUpsampleBilinear2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleBilinear2dOut ¶

func AtgUpsampleBilinear2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleBilinear2dVec ¶ added in v0.8.0

func AtgUpsampleBilinear2dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scaleFactorsData []float64, scaleFactorsLen int)

func AtgUpsampleLinear1d ¶

func AtgUpsampleLinear1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesVal float64, scalesNull int)

func AtgUpsampleLinear1dBackward ¶

func AtgUpsampleLinear1dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesVal float64, scalesNull int)

func AtgUpsampleLinear1dBackwardGradInput ¶ added in v0.4.0

func AtgUpsampleLinear1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesVal float64, scalesNull int)

func AtgUpsampleLinear1dOut ¶

func AtgUpsampleLinear1dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesVal float64, scalesNull int)

func AtgUpsampleLinear1dVec ¶ added in v0.8.0

func AtgUpsampleLinear1dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scaleFactorsData []float64, scaleFactorsLen int)

func AtgUpsampleNearest1d ¶

func AtgUpsampleNearest1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesVal float64, scalesNull int)

func AtgUpsampleNearest1dBackward ¶

func AtgUpsampleNearest1dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesVal float64, scalesNull int)

func AtgUpsampleNearest1dBackwardGradInput ¶ added in v0.4.0

func AtgUpsampleNearest1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesVal float64, scalesNull int)

func AtgUpsampleNearest1dOut ¶

func AtgUpsampleNearest1dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesVal float64, scalesNull int)

func AtgUpsampleNearest1dVec ¶ added in v0.8.0

func AtgUpsampleNearest1dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, scaleFactorsData []float64, scaleFactorsLen int)

func AtgUpsampleNearest2d ¶

func AtgUpsampleNearest2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleNearest2dBackward ¶

func AtgUpsampleNearest2dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleNearest2dBackwardGradInput ¶ added in v0.4.0

func AtgUpsampleNearest2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleNearest2dOut ¶

func AtgUpsampleNearest2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleNearest2dVec ¶ added in v0.8.0

func AtgUpsampleNearest2dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, scaleFactorsData []float64, scaleFactorsLen int)

func AtgUpsampleNearest3d ¶

func AtgUpsampleNearest3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleNearest3dBackward ¶

func AtgUpsampleNearest3dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleNearest3dBackwardGradInput ¶ added in v0.4.0

func AtgUpsampleNearest3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleNearest3dOut ¶

func AtgUpsampleNearest3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleNearest3dVec ¶ added in v0.8.0

func AtgUpsampleNearest3dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, scaleFactorsData []float64, scaleFactorsLen int)

func AtgUpsampleTrilinear3d ¶

func AtgUpsampleTrilinear3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleTrilinear3dBackward ¶

func AtgUpsampleTrilinear3dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleTrilinear3dBackwardGradInput ¶ added in v0.4.0

func AtgUpsampleTrilinear3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleTrilinear3dOut ¶

func AtgUpsampleTrilinear3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func AtgUpsampleTrilinear3dVec ¶ added in v0.8.0

func AtgUpsampleTrilinear3dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scaleFactorsData []float64, scaleFactorsLen int)

func AtgValueSelectingReductionBackward ¶ added in v0.3.0

func AtgValueSelectingReductionBackward(ptr *Ctensor, grad Ctensor, dim int64, indices Ctensor, sizesData []int64, sizesLen int, keepdim int32)

func AtgValues ¶

func AtgValues(ptr *Ctensor, self Ctensor)

func AtgValuesCopy ¶ added in v0.8.0

func AtgValuesCopy(ptr *Ctensor, self Ctensor)

func AtgValuesCopyOut ¶ added in v0.8.0

func AtgValuesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgVander ¶ added in v0.3.0

func AtgVander(ptr *Ctensor, x Ctensor, nVal int64, nNull int, increasing int32)

func AtgVar ¶

func AtgVar(ptr *Ctensor, self Ctensor, unbiased int32)

func AtgVarCorrection ¶ added in v0.4.0

func AtgVarCorrection(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, keepdim int32)

func AtgVarCorrectionOut ¶ added in v0.4.0

func AtgVarCorrectionOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, keepdim int32)

func AtgVarDim ¶ added in v0.4.0

func AtgVarDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, keepdim int32)

func AtgVarMean ¶

func AtgVarMean(ptr *Ctensor, self Ctensor, unbiased int32)

func AtgVarMeanCorrection ¶ added in v0.4.0

func AtgVarMeanCorrection(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, keepdim int32)

func AtgVarMeanCorrectionOut ¶ added in v0.8.0

func AtgVarMeanCorrectionOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, dimData []int64, dimLen int, correction Cscalar, keepdim int32)

func AtgVarMeanDim ¶ added in v0.4.0

func AtgVarMeanDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, keepdim int32)

func AtgVarOut ¶

func AtgVarOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, unbiased int32, keepdim int32)

func AtgVdot ¶ added in v0.3.0

func AtgVdot(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgVdotOut ¶ added in v0.3.0

func AtgVdotOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgView ¶

func AtgView(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgViewAs ¶

func AtgViewAs(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgViewAsComplex ¶ added in v0.3.0

func AtgViewAsComplex(ptr *Ctensor, self Ctensor)

func AtgViewAsComplexCopy ¶ added in v0.8.0

func AtgViewAsComplexCopy(ptr *Ctensor, self Ctensor)

func AtgViewAsComplexCopyOut ¶ added in v0.8.0

func AtgViewAsComplexCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgViewAsReal ¶ added in v0.3.0

func AtgViewAsReal(ptr *Ctensor, self Ctensor)

func AtgViewAsRealCopy ¶ added in v0.8.0

func AtgViewAsRealCopy(ptr *Ctensor, self Ctensor)

func AtgViewAsRealCopyOut ¶ added in v0.8.0

func AtgViewAsRealCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgViewCopy ¶ added in v0.8.0

func AtgViewCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgViewCopyDtype ¶ added in v0.8.0

func AtgViewCopyDtype(ptr *Ctensor, self Ctensor, dtype int32)

func AtgViewCopyDtypeOut ¶ added in v0.8.0

func AtgViewCopyDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32)

func AtgViewCopyOut ¶ added in v0.8.0

func AtgViewCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func AtgViewDtype ¶ added in v0.4.0

func AtgViewDtype(ptr *Ctensor, self Ctensor, dtype int32)

func AtgVstack ¶ added in v0.3.0

func AtgVstack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgVstackOut ¶ added in v0.3.0

func AtgVstackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int)

func AtgWhereScalar ¶ added in v0.4.0

func AtgWhereScalar(ptr *Ctensor, condition Ctensor, selfScalar Cscalar, other Cscalar)

func AtgWhereScalarother ¶ added in v0.4.0

func AtgWhereScalarother(ptr *Ctensor, condition Ctensor, self Ctensor, other Cscalar)

func AtgWhereScalarself ¶ added in v0.4.0

func AtgWhereScalarself(ptr *Ctensor, condition Ctensor, selfScalar Cscalar, other Ctensor)

func AtgWhereSelf ¶ added in v0.4.0

func AtgWhereSelf(ptr *Ctensor, condition Ctensor, self Ctensor, other Ctensor)

func AtgWhereSelfOut ¶ added in v0.8.0

func AtgWhereSelfOut(ptr *Ctensor, out Ctensor, condition Ctensor, self Ctensor, other Ctensor)

func AtgXlogy ¶ added in v0.4.0

func AtgXlogy(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgXlogyOutscalarOther ¶ added in v0.4.0

func AtgXlogyOutscalarOther(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func AtgXlogyOutscalarSelf ¶ added in v0.4.0

func AtgXlogyOutscalarSelf(ptr *Ctensor, out Ctensor, selfScalar Cscalar, other Ctensor)

func AtgXlogyOuttensor ¶ added in v0.4.0

func AtgXlogyOuttensor(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func AtgXlogyScalarOther ¶ added in v0.4.0

func AtgXlogyScalarOther(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgXlogyScalarOther_ ¶ added in v0.4.0

func AtgXlogyScalarOther_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtgXlogyScalarSelf ¶ added in v0.4.0

func AtgXlogyScalarSelf(ptr *Ctensor, selfScalar Cscalar, other Ctensor)

func AtgXlogy_ ¶ added in v0.4.0

func AtgXlogy_(ptr *Ctensor, self Ctensor, other Ctensor)

func AtgZero ¶ added in v0.8.0

func AtgZero(ptr *Ctensor, self Ctensor)

func AtgZeroOut ¶ added in v0.8.0

func AtgZeroOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgZero_ ¶

func AtgZero_(ptr *Ctensor, self Ctensor)

func AtgZeros ¶

func AtgZeros(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func AtgZerosLike ¶

func AtgZerosLike(ptr *Ctensor, self Ctensor)

func AtgZerosLikeOut ¶ added in v0.8.0

func AtgZerosLikeOut(ptr *Ctensor, out Ctensor, self Ctensor)

func AtgZerosOut ¶

func AtgZerosOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)

func Atg_AdaptiveAvgPool2d ¶

func Atg_AdaptiveAvgPool2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func Atg_AdaptiveAvgPool2dBackward ¶

func Atg_AdaptiveAvgPool2dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)

func Atg_AdaptiveAvgPool2dBackwardOut ¶ added in v0.8.0

func Atg_AdaptiveAvgPool2dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor)

func Atg_AdaptiveAvgPool2dOut ¶ added in v0.8.0

func Atg_AdaptiveAvgPool2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func Atg_AdaptiveAvgPool3d ¶ added in v0.4.0

func Atg_AdaptiveAvgPool3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func Atg_AdaptiveAvgPool3dBackward ¶ added in v0.4.0

func Atg_AdaptiveAvgPool3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor)

func Atg_AdaptiveAvgPool3dBackwardOut ¶ added in v0.8.0

func Atg_AdaptiveAvgPool3dBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, self Ctensor)

func Atg_AdaptiveAvgPool3dOut ¶ added in v0.8.0

func Atg_AdaptiveAvgPool3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int)

func Atg_AddBatchDim ¶ added in v0.3.0

func Atg_AddBatchDim(ptr *Ctensor, self Ctensor, batchDim int64, level int64)

func Atg_AddRelu ¶ added in v0.3.0

func Atg_AddRelu(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg_AddReluOut ¶ added in v0.3.0

func Atg_AddReluOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func Atg_AddReluScalar ¶ added in v0.5.0

func Atg_AddReluScalar(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg_AddReluScalarOut ¶ added in v0.8.0

func Atg_AddReluScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func Atg_AddReluScalar_ ¶ added in v0.5.0

func Atg_AddReluScalar_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg_AddRelu_ ¶ added in v0.3.0

func Atg_AddRelu_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg_AddmmActivation ¶ added in v0.8.0

func Atg_AddmmActivation(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor, useGelu int32)

func Atg_AddmmActivationOut ¶ added in v0.8.0

func Atg_AddmmActivationOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor, useGelu int32)

func Atg_Aminmax ¶ added in v0.3.0

func Atg_Aminmax(ptr *Ctensor, self Ctensor)

func Atg_AminmaxDim ¶ added in v0.4.0

func Atg_AminmaxDim(ptr *Ctensor, self Ctensor, dim int64, keepdim int32)

func Atg_AminmaxDimOut ¶ added in v0.8.0

func Atg_AminmaxDimOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, dim int64, keepdim int32)

func Atg_AminmaxOut ¶ added in v0.8.0

func Atg_AminmaxOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor)

func Atg_AmpUpdateScale ¶

func Atg_AmpUpdateScale(ptr *Ctensor, self Ctensor, growthTracker Ctensor, foundInf Ctensor, scaleGrowthFactor float64, scaleBackoffFactor float64, growthInterval int64)

func Atg_AmpUpdateScaleOut ¶ added in v0.8.0

func Atg_AmpUpdateScaleOut(ptr *Ctensor, out Ctensor, self Ctensor, growthTracker Ctensor, foundInf Ctensor, scaleGrowthFactor float64, scaleBackoffFactor float64, growthInterval int64)

func Atg_AmpUpdateScale_ ¶ added in v0.4.0

func Atg_AmpUpdateScale_(ptr *Ctensor, self Ctensor, growthTracker Ctensor, foundInf Ctensor, scaleGrowthFactor float64, scaleBackoffFactor float64, growthInterval int64)

func Atg_AutocastToFullPrecision ¶ added in v0.7.0

func Atg_AutocastToFullPrecision(ptr *Ctensor, self Ctensor, cudaEnabled int32, cpuEnabled int32)

func Atg_AutocastToReducedPrecision ¶ added in v0.7.0

func Atg_AutocastToReducedPrecision(ptr *Ctensor, self Ctensor, cudaEnabled int32, cpuEnabled int32, cudaDtype int32, cpuDtype int32)

func Atg_CastByte ¶

func Atg_CastByte(ptr *Ctensor, self Ctensor, nonBlocking int32)

func Atg_CastChar ¶

func Atg_CastChar(ptr *Ctensor, self Ctensor, nonBlocking int32)

func Atg_CastDouble ¶

func Atg_CastDouble(ptr *Ctensor, self Ctensor, nonBlocking int32)

func Atg_CastFloat ¶

func Atg_CastFloat(ptr *Ctensor, self Ctensor, nonBlocking int32)

func Atg_CastHalf ¶

func Atg_CastHalf(ptr *Ctensor, self Ctensor, nonBlocking int32)

func Atg_CastInt ¶

func Atg_CastInt(ptr *Ctensor, self Ctensor, nonBlocking int32)

func Atg_CastLong ¶

func Atg_CastLong(ptr *Ctensor, self Ctensor, nonBlocking int32)

func Atg_CastShort ¶

func Atg_CastShort(ptr *Ctensor, self Ctensor, nonBlocking int32)

func Atg_CdistBackward ¶

func Atg_CdistBackward(ptr *Ctensor, grad Ctensor, x1 Ctensor, x2 Ctensor, p float64, cdist Ctensor)

func Atg_CdistBackwardOut ¶ added in v0.8.0

func Atg_CdistBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, x1 Ctensor, x2 Ctensor, p float64, cdist Ctensor)

func Atg_CholeskySolveHelper ¶

func Atg_CholeskySolveHelper(ptr *Ctensor, self Ctensor, a Ctensor, upper int32)

func Atg_CholeskySolveHelperOut ¶ added in v0.8.0

func Atg_CholeskySolveHelperOut(ptr *Ctensor, out Ctensor, self Ctensor, a Ctensor, upper int32)

func Atg_Coalesce ¶ added in v0.4.0

func Atg_Coalesce(ptr *Ctensor, self Ctensor)

func Atg_CoalesceOut ¶ added in v0.8.0

func Atg_CoalesceOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_Coalesced ¶ added in v0.8.0

func Atg_Coalesced(ptr *Ctensor, self Ctensor, coalesced int32)

func Atg_CoalescedOut ¶ added in v0.8.0

func Atg_CoalescedOut(ptr *Ctensor, out Ctensor, self Ctensor, coalesced int32)

func Atg_Coalesced_ ¶

func Atg_Coalesced_(ptr *Ctensor, self Ctensor, coalesced int32)

func Atg_ComputeLinearCombination ¶ added in v0.3.0

func Atg_ComputeLinearCombination(ptr *Ctensor, input Ctensor, coefficients Ctensor)

func Atg_ComputeLinearCombinationOut ¶ added in v0.3.0

func Atg_ComputeLinearCombinationOut(ptr *Ctensor, out Ctensor, input Ctensor, coefficients Ctensor)

func Atg_Conj ¶ added in v0.3.0

func Atg_Conj(ptr *Ctensor, self Ctensor)

func Atg_ConjCopy ¶ added in v0.8.0

func Atg_ConjCopy(ptr *Ctensor, self Ctensor)

func Atg_ConjCopyOut ¶ added in v0.8.0

func Atg_ConjCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_ConjPhysical ¶ added in v0.5.0

func Atg_ConjPhysical(ptr *Ctensor, self Ctensor)

func Atg_ConjPhysicalOut ¶ added in v0.8.0

func Atg_ConjPhysicalOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_ConvDepthwise2d ¶ added in v0.5.0

func Atg_ConvDepthwise2d(ptr *Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int)

func Atg_ConvDepthwise2dOut ¶ added in v0.5.0

func Atg_ConvDepthwise2dOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int)

func Atg_ConvertIndicesFromCooToCsr ¶ added in v0.5.0

func Atg_ConvertIndicesFromCooToCsr(ptr *Ctensor, self Ctensor, size int64, outInt32 int32)

func Atg_ConvertIndicesFromCooToCsrOut ¶ added in v0.5.0

func Atg_ConvertIndicesFromCooToCsrOut(ptr *Ctensor, out Ctensor, self Ctensor, size int64, outInt32 int32)

func Atg_ConvertIndicesFromCsrToCoo ¶ added in v0.7.0

func Atg_ConvertIndicesFromCsrToCoo(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, outInt32 int32, transpose int32)

func Atg_ConvertIndicesFromCsrToCooOut ¶ added in v0.7.0

func Atg_ConvertIndicesFromCsrToCooOut(ptr *Ctensor, out Ctensor, crowIndices Ctensor, colIndices Ctensor, outInt32 int32, transpose int32)

func Atg_Convolution ¶

func Atg_Convolution(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, transposed int32, outputPaddingData []int64, outputPaddingLen int, groups int64, benchmark int32, deterministic int32, cudnnEnabled int32, allowTf32 int32)

func Atg_ConvolutionDeprecated ¶ added in v0.4.0

func Atg_ConvolutionDeprecated(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, transposed int32, outputPaddingData []int64, outputPaddingLen int, groups int64, benchmark int32, deterministic int32, cudnnEnabled int32)

func Atg_ConvolutionMode ¶ added in v0.4.0

func Atg_ConvolutionMode(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, padding string, dilationData []int64, dilationLen int, groups int64)

func Atg_ConvolutionOut ¶ added in v0.8.0

func Atg_ConvolutionOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, strideData []int64, strideLen int, paddingData []int64, paddingLen int, dilationData []int64, dilationLen int, transposed int32, outputPaddingData []int64, outputPaddingLen int, groups int64, benchmark int32, deterministic int32, cudnnEnabled int32, allowTf32 int32)

func Atg_CopyFrom ¶

func Atg_CopyFrom(ptr *Ctensor, self Ctensor, dst Ctensor, nonBlocking int32)

func Atg_CopyFromAndResize ¶ added in v0.5.0

func Atg_CopyFromAndResize(ptr *Ctensor, self Ctensor, dst Ctensor)

func Atg_CopyFromAndResizeOut ¶ added in v0.8.0

func Atg_CopyFromAndResizeOut(ptr *Ctensor, out Ctensor, self Ctensor, dst Ctensor)

func Atg_CopyFromOut ¶ added in v0.8.0

func Atg_CopyFromOut(ptr *Ctensor, out Ctensor, self Ctensor, dst Ctensor, nonBlocking int32)

func Atg_CsltCompress ¶ added in v0.9.0

func Atg_CsltCompress(ptr *Ctensor, input Ctensor)

func Atg_CsltSparseMm ¶ added in v0.9.0

func Atg_CsltSparseMm(ptr *Ctensor, compressedA Ctensor, denseB Ctensor, bias Ctensor, transposeResult int32)

func Atg_CtcLoss ¶

func Atg_CtcLoss(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, blank int64, zeroInfinity int32)

func Atg_CtcLossBackward ¶

func Atg_CtcLossBackward(ptr *Ctensor, grad Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, negLogLikelihood Ctensor, logAlpha Ctensor, blank int64, zeroInfinity int32)

func Atg_CtcLossBackwardOut ¶ added in v0.8.0

func Atg_CtcLossBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, negLogLikelihood Ctensor, logAlpha Ctensor, blank int64, zeroInfinity int32)

func Atg_CtcLossBackwardTensor ¶ added in v0.8.0

func Atg_CtcLossBackwardTensor(ptr *Ctensor, grad Ctensor, logProbs Ctensor, targets Ctensor, inputLengths Ctensor, targetLengths Ctensor, negLogLikelihood Ctensor, logAlpha Ctensor, blank int64, zeroInfinity int32)

func Atg_CtcLossOut ¶ added in v0.8.0

func Atg_CtcLossOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, blank int64, zeroInfinity int32)

func Atg_CtcLossTensor ¶ added in v0.8.0

func Atg_CtcLossTensor(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengths Ctensor, targetLengths Ctensor, blank int64, zeroInfinity int32)

func Atg_CtcLossTensorOut ¶ added in v0.8.0

func Atg_CtcLossTensorOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, logProbs Ctensor, targets Ctensor, inputLengths Ctensor, targetLengths Ctensor, blank int64, zeroInfinity int32)

func Atg_CudnnCtcLoss ¶

func Atg_CudnnCtcLoss(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, blank int64, deterministic int32, zeroInfinity int32)

func Atg_CudnnCtcLossOut ¶ added in v0.8.0

func Atg_CudnnCtcLossOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, blank int64, deterministic int32, zeroInfinity int32)

func Atg_CudnnCtcLossTensor ¶ added in v0.8.0

func Atg_CudnnCtcLossTensor(ptr *Ctensor, logProbs Ctensor, targets Ctensor, inputLengths Ctensor, targetLengths Ctensor, blank int64, deterministic int32, zeroInfinity int32)

func Atg_CudnnInitDropoutState ¶

func Atg_CudnnInitDropoutState(ptr *Ctensor, dropout float64, train int32, dropoutSeed int64, optionsKind int32, optionsDevice int32)

func Atg_CudnnInitDropoutStateOut ¶ added in v0.8.0

func Atg_CudnnInitDropoutStateOut(ptr *Ctensor, out Ctensor, dropout float64, train int32, dropoutSeed int64)

func Atg_CudnnRnn ¶

func Atg_CudnnRnn(ptr *Ctensor, input Ctensor, weightData []Ctensor, weightLen int, weightStride0 int64, weightBuf Ctensor, hx Ctensor, cx Ctensor, mode int64, hiddenSize int64, projSize int64, numLayers int64, batchFirst int32, dropout float64, train int32, bidirectional int32, batchSizesData []int64, batchSizesLen int, dropoutState Ctensor)

func Atg_CudnnRnnFlattenWeight ¶

func Atg_CudnnRnnFlattenWeight(ptr *Ctensor, weightArrData []Ctensor, weightArrLen int, weightStride0 int64, inputSize int64, mode int64, hiddenSize int64, projSize int64, numLayers int64, batchFirst int32, bidirectional int32)

func Atg_CudnnRnnFlattenWeightOut ¶ added in v0.8.0

func Atg_CudnnRnnFlattenWeightOut(ptr *Ctensor, out Ctensor, weightArrData []Ctensor, weightArrLen int, weightStride0 int64, inputSize int64, mode int64, hiddenSize int64, projSize int64, numLayers int64, batchFirst int32, bidirectional int32)

func Atg_CudnnRnnOut ¶ added in v0.8.0

func Atg_CudnnRnnOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, out4 Ctensor, input Ctensor, weightData []Ctensor, weightLen int, weightStride0 int64, weightBuf Ctensor, hx Ctensor, cx Ctensor, mode int64, hiddenSize int64, projSize int64, numLayers int64, batchFirst int32, dropout float64, train int32, bidirectional int32, batchSizesData []int64, batchSizesLen int, dropoutState Ctensor)

func Atg_DebugHasInternalOverlap ¶ added in v0.4.0

func Atg_DebugHasInternalOverlap(self Ctensor) int64

func Atg_DimArange ¶

func Atg_DimArange(ptr *Ctensor, like Ctensor, dim int64)

func Atg_Dimi ¶ added in v0.4.0

func Atg_Dimi(self Ctensor) int64

func Atg_Dimv ¶ added in v0.4.0

func Atg_Dimv(self Ctensor) int64

func Atg_DirichletGrad ¶

func Atg_DirichletGrad(ptr *Ctensor, x Ctensor, alpha Ctensor, total Ctensor)

func Atg_DirichletGradOut ¶ added in v0.8.0

func Atg_DirichletGradOut(ptr *Ctensor, out Ctensor, x Ctensor, alpha Ctensor, total Ctensor)

func Atg_EfficientAttentionBackward ¶ added in v0.8.0

func Atg_EfficientAttentionBackward(ptr *Ctensor, gradOut_ Ctensor, query Ctensor, key Ctensor, value Ctensor, bias Ctensor, out Ctensor, cuSeqlensQ Ctensor, cuSeqlensK Ctensor, maxSeqlenK int64, maxSeqlenQ int64, logsumexp Ctensor, dropoutP float64, philoxSeed Ctensor, philoxOffset Ctensor, customMaskType int64, biasRequiresGrad int32, scaleVal float64, scaleNull int, numSplitsKeyVal int64, numSplitsKeyNull int)

func Atg_Efficientzerotensor ¶ added in v0.7.0

func Atg_Efficientzerotensor(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func Atg_EfficientzerotensorOut ¶ added in v0.8.0

func Atg_EfficientzerotensorOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int)

func Atg_EmbeddingBag ¶

func Atg_EmbeddingBag(ptr *Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, scaleGradByFreq int32, mode int64, sparse int32, perSampleWeights Ctensor, includeLastOffset int32, paddingIdx int64)

func Atg_EmbeddingBagBackward ¶

func Atg_EmbeddingBagBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, offsets Ctensor, offset2bag Ctensor, bagSize Ctensor, maximumIndices Ctensor, numWeights int64, scaleGradByFreq int32, mode int64, sparse int32, perSampleWeights Ctensor, paddingIdx int64)

func Atg_EmbeddingBagDenseBackward ¶

func Atg_EmbeddingBagDenseBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, offset2bag Ctensor, bagSize Ctensor, maximumIndices Ctensor, numWeights int64, scaleGradByFreq int32, mode int64, perSampleWeights Ctensor, paddingIdx int64)

func Atg_EmbeddingBagDenseBackwardOut ¶ added in v0.8.0

func Atg_EmbeddingBagDenseBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, indices Ctensor, offset2bag Ctensor, bagSize Ctensor, maximumIndices Ctensor, numWeights int64, scaleGradByFreq int32, mode int64, perSampleWeights Ctensor, paddingIdx int64)

func Atg_EmbeddingBagForwardOnly ¶ added in v0.3.0

func Atg_EmbeddingBagForwardOnly(ptr *Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, scaleGradByFreq int32, mode int64, sparse int32, perSampleWeights Ctensor, includeLastOffset int32, paddingIdx int64)

func Atg_EmbeddingBagForwardOnlyOut ¶ added in v0.8.0

func Atg_EmbeddingBagForwardOnlyOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, scaleGradByFreq int32, mode int64, sparse int32, perSampleWeights Ctensor, includeLastOffset int32, paddingIdx int64)

func Atg_EmbeddingBagOut ¶ added in v0.8.0

func Atg_EmbeddingBagOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, scaleGradByFreq int32, mode int64, sparse int32, perSampleWeights Ctensor, includeLastOffset int32, paddingIdx int64)

func Atg_EmbeddingBagPerSampleWeightsBackward ¶

func Atg_EmbeddingBagPerSampleWeightsBackward(ptr *Ctensor, grad Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, offset2bag Ctensor, mode int64, paddingIdx int64)

func Atg_EmbeddingBagPerSampleWeightsBackwardOut ¶ added in v0.8.0

func Atg_EmbeddingBagPerSampleWeightsBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, weight Ctensor, indices Ctensor, offsets Ctensor, offset2bag Ctensor, mode int64, paddingIdx int64)

func Atg_EmbeddingBagSparseBackward ¶

func Atg_EmbeddingBagSparseBackward(ptr *Ctensor, grad Ctensor, indices Ctensor, offsets Ctensor, offset2bag Ctensor, bagSize Ctensor, numWeights int64, scaleGradByFreq int32, mode int64, perSampleWeights Ctensor, paddingIdx int64)

func Atg_EmptyAffineQuantized ¶

func Atg_EmptyAffineQuantized(ptr *Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32, scale float64, zeroPoint int64)

func Atg_EmptyAffineQuantizedOut ¶ added in v0.8.0

func Atg_EmptyAffineQuantizedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, scale float64, zeroPoint int64)

func Atg_EmptyPerChannelAffineQuantized ¶

func Atg_EmptyPerChannelAffineQuantized(ptr *Ctensor, sizeData []int64, sizeLen int, scales Ctensor, zeroPoints Ctensor, axis int64, optionsKind int32, optionsDevice int32)

func Atg_EmptyPerChannelAffineQuantizedOut ¶ added in v0.8.0

func Atg_EmptyPerChannelAffineQuantizedOut(ptr *Ctensor, out Ctensor, sizeData []int64, sizeLen int, scales Ctensor, zeroPoints Ctensor, axis int64)

func Atg_EuclideanDist ¶ added in v0.3.0

func Atg_EuclideanDist(ptr *Ctensor, x1 Ctensor, x2 Ctensor)

func Atg_EuclideanDistOut ¶ added in v0.8.0

func Atg_EuclideanDistOut(ptr *Ctensor, out Ctensor, x1 Ctensor, x2 Ctensor)

func Atg_FakeQuantizeLearnablePerChannelAffine ¶ added in v0.3.0

func Atg_FakeQuantizeLearnablePerChannelAffine(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64, quantMin int64, quantMax int64, gradFactor float64)

func Atg_FakeQuantizeLearnablePerChannelAffineBackward ¶ added in v0.3.0

func Atg_FakeQuantizeLearnablePerChannelAffineBackward(ptr *Ctensor, grad Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64, quantMin int64, quantMax int64, gradFactor float64)

func Atg_FakeQuantizeLearnablePerChannelAffineOut ¶ added in v0.8.0

func Atg_FakeQuantizeLearnablePerChannelAffineOut(ptr *Ctensor, out Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64, quantMin int64, quantMax int64, gradFactor float64)

func Atg_FakeQuantizeLearnablePerTensorAffine ¶ added in v0.3.0

func Atg_FakeQuantizeLearnablePerTensorAffine(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, quantMin int64, quantMax int64, gradFactor float64)

func Atg_FakeQuantizeLearnablePerTensorAffineBackward ¶ added in v0.3.0

func Atg_FakeQuantizeLearnablePerTensorAffineBackward(ptr *Ctensor, grad Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, quantMin int64, quantMax int64, gradFactor float64)

func Atg_FakeQuantizeLearnablePerTensorAffineOut ¶ added in v0.8.0

func Atg_FakeQuantizeLearnablePerTensorAffineOut(ptr *Ctensor, out Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, quantMin int64, quantMax int64, gradFactor float64)

func Atg_FakeQuantizePerTensorAffineCachemaskTensorQparams ¶ added in v0.5.0

func Atg_FakeQuantizePerTensorAffineCachemaskTensorQparams(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, fakeQuantEnabled Ctensor, quantMin int64, quantMax int64)

func Atg_FakeQuantizePerTensorAffineCachemaskTensorQparamsOut ¶ added in v0.8.0

func Atg_FakeQuantizePerTensorAffineCachemaskTensorQparamsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, fakeQuantEnabled Ctensor, quantMin int64, quantMax int64)

func Atg_FftC2c ¶ added in v0.4.0

func Atg_FftC2c(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, normalization int64, forward int32)

func Atg_FftC2cOut ¶ added in v0.4.0

func Atg_FftC2cOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, normalization int64, forward int32)

func Atg_FftC2r ¶ added in v0.4.0

func Atg_FftC2r(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, normalization int64, lastDimSize int64)

func Atg_FftC2rOut ¶ added in v0.4.0

func Atg_FftC2rOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, normalization int64, lastDimSize int64)

func Atg_FftR2c ¶ added in v0.4.0

func Atg_FftR2c(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, normalization int64, onesided int32)

func Atg_FftR2cOut ¶ added in v0.4.0

func Atg_FftR2cOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, normalization int64, onesided int32)

func Atg_FillMemEffDropoutMask_ ¶ added in v0.9.0

func Atg_FillMemEffDropoutMask_(ptr *Ctensor, self Ctensor, dropoutP float64, seed int64, offset int64)

func Atg_FlashAttentionBackward ¶ added in v0.8.0

func Atg_FlashAttentionBackward(ptr *Ctensor, gradOut Ctensor, query Ctensor, key Ctensor, value Ctensor, out Ctensor, logsumexp Ctensor, cumSeqQ Ctensor, cumSeqK Ctensor, maxQ int64, maxK int64, dropoutP float64, isCausal int32, philoxSeed Ctensor, philoxOffset Ctensor, scaleVal float64, scaleNull int)

func Atg_Foobar ¶ added in v0.8.0

func Atg_Foobar(ptr *Ctensor, self Ctensor, arg1 int32, arg2 int32, arg3 int32)

func Atg_FoobarOut ¶ added in v0.8.0

func Atg_FoobarOut(ptr *Ctensor, out Ctensor, self Ctensor, arg1 int32, arg2 int32, arg3 int32)

func Atg_FunctionalAssertAsync ¶ added in v0.9.0

func Atg_FunctionalAssertAsync(ptr *Ctensor, self Ctensor, assertMsg string, depToken Ctensor)

func Atg_FunctionalSymConstrainRange ¶ added in v0.9.0

func Atg_FunctionalSymConstrainRange(ptr *Ctensor, size Cscalar, minVal int64, minNull int, maxVal int64, maxNull int, depToken Ctensor)

func Atg_FunctionalSymConstrainRangeForSize ¶ added in v0.9.0

func Atg_FunctionalSymConstrainRangeForSize(ptr *Ctensor, size Cscalar, minVal int64, minNull int, maxVal int64, maxNull int, depToken Ctensor)

func Atg_FusedDropout ¶

func Atg_FusedDropout(ptr *Ctensor, self Ctensor, p float64)

func Atg_FusedDropoutOut ¶ added in v0.8.0

func Atg_FusedDropoutOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, p float64)

func Atg_FusedMovingAvgObsFqHelper ¶ added in v0.5.0

func Atg_FusedMovingAvgObsFqHelper(ptr *Ctensor, self Ctensor, observerOn Ctensor, fakeQuantOn Ctensor, runningMin Ctensor, runningMax Ctensor, scale Ctensor, zeroPoint Ctensor, averagingConst float64, quantMin int64, quantMax int64, chAxis int64, perRowFakeQuant int32, symmetricQuant int32)

func Atg_FusedMovingAvgObsFqHelperFunctional ¶ added in v0.8.0

func Atg_FusedMovingAvgObsFqHelperFunctional(ptr *Ctensor, self Ctensor, observerOn Ctensor, fakeQuantOn Ctensor, runningMin Ctensor, runningMax Ctensor, scale Ctensor, zeroPoint Ctensor, averagingConst float64, quantMin int64, quantMax int64, chAxis int64, perRowFakeQuant int32, symmetricQuant int32)

func Atg_FusedMovingAvgObsFqHelperOut ¶ added in v0.8.0

func Atg_FusedMovingAvgObsFqHelperOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, observerOn Ctensor, fakeQuantOn Ctensor, runningMin Ctensor, runningMax Ctensor, scale Ctensor, zeroPoint Ctensor, averagingConst float64, quantMin int64, quantMax int64, chAxis int64, perRowFakeQuant int32, symmetricQuant int32)

func Atg_FusedSdpChoice ¶ added in v0.8.0

func Atg_FusedSdpChoice(query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int) int64

func Atg_FwPrimal ¶ added in v0.4.0

func Atg_FwPrimal(ptr *Ctensor, self Ctensor, level int64)

func Atg_FwPrimalCopy ¶ added in v0.8.0

func Atg_FwPrimalCopy(ptr *Ctensor, self Ctensor, level int64)

func Atg_FwPrimalCopyOut ¶ added in v0.8.0

func Atg_FwPrimalCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, level int64)

func Atg_GatherSparseBackward ¶

func Atg_GatherSparseBackward(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, grad Ctensor)

func Atg_GridSampler2dCpuFallback ¶ added in v0.3.0

func Atg_GridSampler2dCpuFallback(ptr *Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, paddingMode int64, alignCorners int32)

func Atg_GridSampler2dCpuFallbackBackward ¶ added in v0.3.0

func Atg_GridSampler2dCpuFallbackBackward(ptr *Ctensor, gradOutput Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, paddingMode int64, alignCorners int32)

func Atg_GridSampler2dCpuFallbackOut ¶ added in v0.8.0

func Atg_GridSampler2dCpuFallbackOut(ptr *Ctensor, out Ctensor, input Ctensor, grid Ctensor, interpolationMode int64, paddingMode int64, alignCorners int32)

func Atg_HasCompatibleShallowCopyType ¶ added in v0.4.0

func Atg_HasCompatibleShallowCopyType(self Ctensor, from Ctensor) bool

func Atg_HasSameStorageNumel ¶ added in v0.7.0

func Atg_HasSameStorageNumel(self Ctensor, other Ctensor) bool

func Atg_HistogramddFromBinCts ¶ added in v0.8.0

func Atg_HistogramddFromBinCts(ptr *Ctensor, out Ctensor, self Ctensor, binsData []int64, binsLen int, rangeValsData []float64, rangeValsLen int, weight Ctensor, density int32)

func Atg_HistogramddFromBinTensors ¶ added in v0.7.0

func Atg_HistogramddFromBinTensors(ptr *Ctensor, self Ctensor, binsData []Ctensor, binsLen int, weight Ctensor, density int32)

func Atg_HistogramddFromBinTensorsOut ¶ added in v0.8.0

func Atg_HistogramddFromBinTensorsOut(ptr *Ctensor, out Ctensor, self Ctensor, binsData []Ctensor, binsLen int, weight Ctensor, density int32)

func Atg_IndexPutImpl ¶ added in v0.8.0

func Atg_IndexPutImpl(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32, unsafety int32)

func Atg_IndexPutImplOut ¶ added in v0.8.0

func Atg_IndexPutImplOut(ptr *Ctensor, out Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32, unsafety int32)

func Atg_IndexPutImpl_ ¶

func Atg_IndexPutImpl_(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32, unsafety int32)

func Atg_Indices ¶

func Atg_Indices(ptr *Ctensor, self Ctensor)

func Atg_IndicesCopy ¶ added in v0.8.0

func Atg_IndicesCopy(ptr *Ctensor, self Ctensor)

func Atg_IndicesCopyOut ¶ added in v0.8.0

func Atg_IndicesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_IntMm ¶ added in v0.9.0

func Atg_IntMm(ptr *Ctensor, self Ctensor, mat2 Ctensor)

func Atg_IntMmOut ¶ added in v0.9.0

func Atg_IntMmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat2 Ctensor)

func Atg_IsAllTrue ¶ added in v0.8.0

func Atg_IsAllTrue(ptr *Ctensor, self Ctensor)

func Atg_IsAnyTrue ¶ added in v0.8.0

func Atg_IsAnyTrue(ptr *Ctensor, self Ctensor)

func Atg_IsZerotensor ¶ added in v0.7.0

func Atg_IsZerotensor(self Ctensor) bool

func Atg_LinalgDet ¶ added in v0.8.0

func Atg_LinalgDet(ptr *Ctensor, a Ctensor)

func Atg_LinalgDetResult ¶ added in v0.8.0

func Atg_LinalgDetResult(ptr *Ctensor, result Ctensor, lU Ctensor, pivots Ctensor, a Ctensor)

func Atg_LinalgEigh ¶ added in v0.8.0

func Atg_LinalgEigh(ptr *Ctensor, a Ctensor, uPLO string, computeV int32)

func Atg_LinalgEighEigenvalues ¶ added in v0.8.0

func Atg_LinalgEighEigenvalues(ptr *Ctensor, eigenvalues Ctensor, eigenvectors Ctensor, a Ctensor, uPLO string, computeV int32)

func Atg_LinalgSlogdet ¶ added in v0.8.0

func Atg_LinalgSlogdet(ptr *Ctensor, a Ctensor)

func Atg_LinalgSlogdetSign ¶ added in v0.8.0

func Atg_LinalgSlogdetSign(ptr *Ctensor, sign Ctensor, logabsdet Ctensor, lU Ctensor, pivots Ctensor, a Ctensor)

func Atg_LinalgSolveEx ¶ added in v0.8.0

func Atg_LinalgSolveEx(ptr *Ctensor, a Ctensor, b Ctensor, left int32, checkErrors int32)

func Atg_LinalgSolveExResult ¶ added in v0.8.0

func Atg_LinalgSolveExResult(ptr *Ctensor, result Ctensor, lU Ctensor, pivots Ctensor, info Ctensor, a Ctensor, b Ctensor, left int32, checkErrors int32)

func Atg_LinalgSvd ¶ added in v0.7.0

func Atg_LinalgSvd(ptr *Ctensor, a Ctensor, fullMatrices int32, computeUv int32, driver string)

func Atg_LinalgSvdU ¶ added in v0.7.0

func Atg_LinalgSvdU(ptr *Ctensor, u Ctensor, s Ctensor, vh Ctensor, a Ctensor, fullMatrices int32, computeUv int32, driver string)

func Atg_LogSoftmax ¶

func Atg_LogSoftmax(ptr *Ctensor, self Ctensor, dim int64, halfToFloat int32)

func Atg_LogSoftmaxBackwardData ¶

func Atg_LogSoftmaxBackwardData(ptr *Ctensor, gradOutput Ctensor, output Ctensor, dim int64, inputDtype int32)

func Atg_LogSoftmaxBackwardDataOut ¶ added in v0.5.0

func Atg_LogSoftmaxBackwardDataOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, dim int64, inputDtype int32)

func Atg_LogSoftmaxOut ¶ added in v0.5.0

func Atg_LogSoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, halfToFloat int32)

func Atg_Logcumsumexp ¶ added in v0.3.0

func Atg_Logcumsumexp(ptr *Ctensor, self Ctensor, dim int64)

func Atg_LogcumsumexpOut ¶ added in v0.3.0

func Atg_LogcumsumexpOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64)

func Atg_LstmMps ¶ added in v0.8.0

func Atg_LstmMps(ptr *Ctensor, input Ctensor, hxData []Ctensor, hxLen int, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32, batchFirst int32)

func Atg_LstmMpsOut ¶ added in v0.8.0

func Atg_LstmMpsOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, out3 Ctensor, out4 Ctensor, out5 Ctensor, input Ctensor, hxData []Ctensor, hxLen int, paramsData []Ctensor, paramsLen int, hasBiases int32, numLayers int64, dropout float64, train int32, bidirectional int32, batchFirst int32)

func Atg_LuWithInfo ¶

func Atg_LuWithInfo(ptr *Ctensor, self Ctensor, pivot int32, checkErrors int32)

func Atg_MakeDepToken ¶ added in v0.9.0

func Atg_MakeDepToken(ptr *Ctensor, optionsKind int32, optionsDevice int32)

func Atg_MakeDual ¶ added in v0.4.0

func Atg_MakeDual(ptr *Ctensor, primal Ctensor, tangent Ctensor, level int64)

func Atg_MakeDualCopy ¶ added in v0.8.0

func Atg_MakeDualCopy(ptr *Ctensor, primal Ctensor, tangent Ctensor, level int64)

func Atg_MakeDualCopyOut ¶ added in v0.8.0

func Atg_MakeDualCopyOut(ptr *Ctensor, out Ctensor, primal Ctensor, tangent Ctensor, level int64)

func Atg_MakePerChannelQuantizedTensor ¶

func Atg_MakePerChannelQuantizedTensor(ptr *Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64)

func Atg_MakePerChannelQuantizedTensorOut ¶ added in v0.8.0

func Atg_MakePerChannelQuantizedTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, scale Ctensor, zeroPoint Ctensor, axis int64)

func Atg_MakePerTensorQuantizedTensor ¶

func Atg_MakePerTensorQuantizedTensor(ptr *Ctensor, self Ctensor, scale float64, zeroPoint int64)

func Atg_MakePerTensorQuantizedTensorOut ¶ added in v0.8.0

func Atg_MakePerTensorQuantizedTensorOut(ptr *Ctensor, out Ctensor, self Ctensor, scale float64, zeroPoint int64)

func Atg_MaskedScale ¶

func Atg_MaskedScale(ptr *Ctensor, self Ctensor, mask Ctensor, scale float64)

func Atg_MaskedScaleOut ¶ added in v0.8.0

func Atg_MaskedScaleOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, scale float64)

func Atg_MaskedSoftmax ¶ added in v0.7.0

func Atg_MaskedSoftmax(ptr *Ctensor, self Ctensor, mask Ctensor, dimVal int64, dimNull int, maskTypeVal int64, maskTypeNull int)

func Atg_MaskedSoftmaxBackward ¶ added in v0.8.0

func Atg_MaskedSoftmaxBackward(ptr *Ctensor, gradOutput Ctensor, output Ctensor, mask Ctensor, dimVal int64, dimNull int)

func Atg_MaskedSoftmaxBackwardOut ¶ added in v0.8.0

func Atg_MaskedSoftmaxBackwardOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, mask Ctensor, dimVal int64, dimNull int)

func Atg_MaskedSoftmaxOut ¶ added in v0.8.0

func Atg_MaskedSoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, dimVal int64, dimNull int, maskTypeVal int64, maskTypeNull int)

func Atg_MkldnnReshape ¶

func Atg_MkldnnReshape(ptr *Ctensor, self Ctensor, shapeData []int64, shapeLen int)

func Atg_MkldnnReshapeOut ¶ added in v0.8.0

func Atg_MkldnnReshapeOut(ptr *Ctensor, out Ctensor, self Ctensor, shapeData []int64, shapeLen int)

func Atg_MkldnnTranspose ¶

func Atg_MkldnnTranspose(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)

func Atg_MkldnnTransposeOut ¶ added in v0.8.0

func Atg_MkldnnTransposeOut(ptr *Ctensor, out Ctensor, self Ctensor, dim0 int64, dim1 int64)

func Atg_MkldnnTranspose_ ¶

func Atg_MkldnnTranspose_(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)

func Atg_MpsConvolution ¶ added in v0.8.0

func Atg_MpsConvolution(ptr *Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64)

func Atg_MpsConvolutionOut ¶ added in v0.8.0

func Atg_MpsConvolutionOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64)

func Atg_MpsConvolutionTranspose ¶ added in v0.8.0

func Atg_MpsConvolutionTranspose(ptr *Ctensor, self Ctensor, weight Ctensor, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64)

func Atg_MpsConvolutionTransposeOut ¶ added in v0.8.0

func Atg_MpsConvolutionTransposeOut(ptr *Ctensor, out Ctensor, self Ctensor, weight Ctensor, paddingData []int64, paddingLen int, outputPaddingData []int64, outputPaddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64)

func Atg_NativeBatchNormLegit ¶ added in v0.8.0

func Atg_NativeBatchNormLegit(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, momentum float64, eps float64)

func Atg_NativeBatchNormLegitFunctional ¶ added in v0.8.0

func Atg_NativeBatchNormLegitFunctional(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, momentum float64, eps float64)

func Atg_NativeBatchNormLegitNoStats ¶ added in v0.8.0

func Atg_NativeBatchNormLegitNoStats(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, training int32, momentum float64, eps float64)

func Atg_NativeBatchNormLegitNoStatsOut ¶ added in v0.8.0

func Atg_NativeBatchNormLegitNoStatsOut(ptr *Ctensor, out Ctensor, saveMean Ctensor, saveInvstd Ctensor, input Ctensor, weight Ctensor, bias Ctensor, training int32, momentum float64, eps float64)

func Atg_NativeBatchNormLegitNoTraining ¶ added in v0.9.0

func Atg_NativeBatchNormLegitNoTraining(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64)

func Atg_NativeBatchNormLegitNoTrainingOut ¶ added in v0.9.0

func Atg_NativeBatchNormLegitNoTrainingOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, momentum float64, eps float64)

func Atg_NativeBatchNormLegitOut ¶ added in v0.8.0

func Atg_NativeBatchNormLegitOut(ptr *Ctensor, out Ctensor, saveMean Ctensor, saveInvstd Ctensor, input Ctensor, weight Ctensor, bias Ctensor, runningMean Ctensor, runningVar Ctensor, training int32, momentum float64, eps float64)

func Atg_NativeMultiHeadAttention ¶ added in v0.8.0

func Atg_NativeMultiHeadAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, embedDim int64, numHead int64, qkvWeight Ctensor, qkvBias Ctensor, projWeight Ctensor, projBias Ctensor, mask Ctensor, needWeights int32, averageAttnWeights int32, maskTypeVal int64, maskTypeNull int)

func Atg_NativeMultiHeadAttentionOut ¶ added in v0.8.0

func Atg_NativeMultiHeadAttentionOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, query Ctensor, key Ctensor, value Ctensor, embedDim int64, numHead int64, qkvWeight Ctensor, qkvBias Ctensor, projWeight Ctensor, projBias Ctensor, mask Ctensor, needWeights int32, averageAttnWeights int32, maskTypeVal int64, maskTypeNull int)

func Atg_NegView ¶ added in v0.5.0

func Atg_NegView(ptr *Ctensor, self Ctensor)

func Atg_NegViewCopy ¶ added in v0.8.0

func Atg_NegViewCopy(ptr *Ctensor, self Ctensor)

func Atg_NegViewCopyOut ¶ added in v0.8.0

func Atg_NegViewCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_NestedFromPadded ¶ added in v0.8.0

func Atg_NestedFromPadded(ptr *Ctensor, padded Ctensor, cpuNestedShapeExample Ctensor, fuseTransform0213 int32)

func Atg_NestedFromPaddedAndNestedExample ¶ added in v0.8.0

func Atg_NestedFromPaddedAndNestedExample(ptr *Ctensor, padded Ctensor, ntExample Ctensor)

func Atg_NestedFromPaddedAndNestedExampleOut ¶ added in v0.8.0

func Atg_NestedFromPaddedAndNestedExampleOut(ptr *Ctensor, out Ctensor, padded Ctensor, ntExample Ctensor)

func Atg_NestedFromPaddedOut ¶ added in v0.8.0

func Atg_NestedFromPaddedOut(ptr *Ctensor, out Ctensor, padded Ctensor, cpuNestedShapeExample Ctensor, fuseTransform0213 int32)

func Atg_NestedSelectBackward ¶ added in v0.8.0

func Atg_NestedSelectBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, dim int64, index int64)

func Atg_NestedSumBackward ¶ added in v0.8.0

func Atg_NestedSumBackward(ptr *Ctensor, grad Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32)

func Atg_NestedViewFromBuffer ¶ added in v0.8.0

func Atg_NestedViewFromBuffer(ptr *Ctensor, self Ctensor, nestedSize Ctensor, nestedStrides Ctensor, offsets Ctensor)

func Atg_NestedViewFromBufferCopy ¶ added in v0.8.0

func Atg_NestedViewFromBufferCopy(ptr *Ctensor, self Ctensor, nestedSize Ctensor, nestedStrides Ctensor, offsets Ctensor)

func Atg_NestedViewFromBufferCopyOut ¶ added in v0.8.0

func Atg_NestedViewFromBufferCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, nestedSize Ctensor, nestedStrides Ctensor, offsets Ctensor)

func Atg_NewZerosWithSameFeatureMeta ¶ added in v0.7.0

func Atg_NewZerosWithSameFeatureMeta(ptr *Ctensor, self Ctensor, other Ctensor, selfNumBatchDims int64)

func Atg_NewZerosWithSameFeatureMetaOut ¶ added in v0.8.0

func Atg_NewZerosWithSameFeatureMetaOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor, selfNumBatchDims int64)

func Atg_NnpackAvailable ¶ added in v0.4.0

func Atg_NnpackAvailable() bool

func Atg_NnpackSpatialConvolution ¶

func Atg_NnpackSpatialConvolution(ptr *Ctensor, input Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int)

func Atg_NnpackSpatialConvolutionOut ¶ added in v0.8.0

func Atg_NnpackSpatialConvolutionOut(ptr *Ctensor, out Ctensor, input Ctensor, weight Ctensor, bias Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int)

func Atg_Nnz ¶ added in v0.4.0

func Atg_Nnz(self Ctensor) int64

func Atg_PackPaddedSequence ¶

func Atg_PackPaddedSequence(ptr *Ctensor, input Ctensor, lengths Ctensor, batchFirst int32)

func Atg_PackPaddedSequenceBackward ¶

func Atg_PackPaddedSequenceBackward(ptr *Ctensor, grad Ctensor, inputSizeData []int64, inputSizeLen int, batchSizes Ctensor, batchFirst int32)

func Atg_PackPaddedSequenceOut ¶ added in v0.8.0

func Atg_PackPaddedSequenceOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, input Ctensor, lengths Ctensor, batchFirst int32)

func Atg_PadCircular ¶ added in v0.8.0

func Atg_PadCircular(ptr *Ctensor, self Ctensor, padData []int64, padLen int)

func Atg_PadEnum ¶ added in v0.8.0

func Atg_PadEnum(ptr *Ctensor, self Ctensor, padData []int64, padLen int, mode int64, valueVal float64, valueNull int)

func Atg_PadPackedSequence ¶

func Atg_PadPackedSequence(ptr *Ctensor, data Ctensor, batchSizes Ctensor, batchFirst int32, paddingValue Cscalar, totalLength int64)

func Atg_PdistBackward ¶

func Atg_PdistBackward(ptr *Ctensor, grad Ctensor, self Ctensor, p float64, pdist Ctensor)

func Atg_PdistBackwardOut ¶ added in v0.8.0

func Atg_PdistBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, self Ctensor, p float64, pdist Ctensor)

func Atg_PinMemory ¶ added in v0.5.0

func Atg_PinMemory(ptr *Ctensor, self Ctensor, device int32)

func Atg_PinMemoryOut ¶ added in v0.8.0

func Atg_PinMemoryOut(ptr *Ctensor, out Ctensor, self Ctensor, device int32)

func Atg_PreluKernel ¶ added in v0.8.0

func Atg_PreluKernel(ptr *Ctensor, self Ctensor, weight Ctensor)

func Atg_PreluKernelBackward ¶ added in v0.8.0

func Atg_PreluKernelBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, weight Ctensor)

func Atg_RemoveBatchDim ¶ added in v0.3.0

func Atg_RemoveBatchDim(ptr *Ctensor, self Ctensor, level int64, batchSize int64, outDim int64)

func Atg_ReshapeAlias ¶ added in v0.5.0

func Atg_ReshapeAlias(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int)

func Atg_ReshapeAliasCopy ¶ added in v0.8.0

func Atg_ReshapeAliasCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int)

func Atg_ReshapeAliasCopyOut ¶ added in v0.8.0

func Atg_ReshapeAliasCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, strideData []int64, strideLen int)

func Atg_ReshapeCopy ¶ added in v0.8.0

func Atg_ReshapeCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func Atg_ReshapeFromTensor ¶

func Atg_ReshapeFromTensor(ptr *Ctensor, self Ctensor, shape Ctensor)

func Atg_ResizeOutput ¶ added in v0.8.0

func Atg_ResizeOutput(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, device int32)

func Atg_ResizeOutputOut ¶ added in v0.8.0

func Atg_ResizeOutputOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int, device int32)

func Atg_ResizeOutput_ ¶ added in v0.8.0

func Atg_ResizeOutput_(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int, device int32)

func Atg_RowwisePrune ¶ added in v0.4.0

func Atg_RowwisePrune(ptr *Ctensor, weight Ctensor, mask Ctensor, compressedIndicesDtype int32)

func Atg_SampleDirichlet ¶

func Atg_SampleDirichlet(ptr *Ctensor, self Ctensor)

func Atg_SampleDirichletOut ¶ added in v0.8.0

func Atg_SampleDirichletOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_SaturateWeightToFp16 ¶ added in v0.3.0

func Atg_SaturateWeightToFp16(ptr *Ctensor, weight Ctensor)

func Atg_ScaledDotProductAttentionMath ¶ added in v0.8.0

func Atg_ScaledDotProductAttentionMath(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnMask Ctensor, dropoutP float64, isCausal int32, dropoutMask Ctensor, scaleVal float64, scaleNull int)

func Atg_ScaledDotProductEfficientAttention ¶ added in v0.8.0

func Atg_ScaledDotProductEfficientAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, attnBias Ctensor, computeLogSumexp int32, dropoutP float64, isCausal int32, scaleVal float64, scaleNull int)

func Atg_ScaledDotProductFlashAttentionBackward ¶ added in v0.8.0

func Atg_ScaledDotProductFlashAttentionBackward(ptr *Ctensor, gradOut Ctensor, query Ctensor, key Ctensor, value Ctensor, out Ctensor, logsumexp Ctensor, cumSeqQ Ctensor, cumSeqK Ctensor, maxQ int64, maxK int64, dropoutP float64, isCausal int32, philoxSeed Ctensor, philoxOffset Ctensor, scaleVal float64, scaleNull int)

func Atg_ScaledMm ¶ added in v0.9.0

func Atg_ScaledMm(ptr *Ctensor, self Ctensor, mat2 Ctensor, bias Ctensor, outDtype int32, scaleA Ctensor, scaleB Ctensor, scaleResult Ctensor)

func Atg_ScaledMmOut ¶ added in v0.9.0

func Atg_ScaledMmOut(ptr *Ctensor, out Ctensor, outAmax Ctensor, self Ctensor, mat2 Ctensor, bias Ctensor, outDtype int32, scaleA Ctensor, scaleB Ctensor, scaleResult Ctensor)

func Atg_ScatterReduce ¶ added in v0.9.0

func Atg_ScatterReduce(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, reduce string, includeSelf int32)

func Atg_ScatterReduceTwoOut ¶ added in v0.9.0

func Atg_ScatterReduceTwoOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, reduce string, includeSelf int32)

func Atg_ScatterReduce_ ¶ added in v0.9.0

func Atg_ScatterReduce_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, src Ctensor, reduce string, includeSelf int32)

func Atg_SegmentReduceBackward ¶ added in v0.5.0

func Atg_SegmentReduceBackward(ptr *Ctensor, grad Ctensor, output Ctensor, data Ctensor, reduce string, lengths Ctensor, offsets Ctensor, axis int64, initial Cscalar)

func Atg_SegmentReduceBackwardOut ¶ added in v0.8.0

func Atg_SegmentReduceBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, output Ctensor, data Ctensor, reduce string, lengths Ctensor, offsets Ctensor, axis int64, initial Cscalar)

func Atg_ShapeAsTensor ¶

func Atg_ShapeAsTensor(ptr *Ctensor, self Ctensor)

func Atg_SlowConv2dBackward ¶ added in v0.7.0

func Atg_SlowConv2dBackward(ptr *Ctensor, gradInput Ctensor, gradWeight Ctensor, gradBias Ctensor, gradOutput Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int)

func Atg_SobolEngineDraw ¶

func Atg_SobolEngineDraw(ptr *Ctensor, quasi Ctensor, n int64, sobolstate Ctensor, dimension int64, numGenerated int64, dtype int32)

func Atg_SobolEngineFf_ ¶

func Atg_SobolEngineFf_(ptr *Ctensor, self Ctensor, n int64, sobolstate Ctensor, dimension int64, numGenerated int64)

func Atg_SobolEngineInitializeState_ ¶

func Atg_SobolEngineInitializeState_(ptr *Ctensor, self Ctensor, dimension int64)

func Atg_SobolEngineScramble_ ¶

func Atg_SobolEngineScramble_(ptr *Ctensor, self Ctensor, ltm Ctensor, dimension int64)

func Atg_Softmax ¶

func Atg_Softmax(ptr *Ctensor, self Ctensor, dim int64, halfToFloat int32)

func Atg_SoftmaxBackwardData ¶

func Atg_SoftmaxBackwardData(ptr *Ctensor, gradOutput Ctensor, output Ctensor, dim int64, inputDtype int32)

func Atg_SoftmaxBackwardDataOut ¶ added in v0.5.0

func Atg_SoftmaxBackwardDataOut(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, output Ctensor, dim int64, inputDtype int32)

func Atg_SoftmaxOut ¶ added in v0.5.0

func Atg_SoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, halfToFloat int32)

func Atg_SparseAddmm ¶

func Atg_SparseAddmm(ptr *Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)

func Atg_SparseAddmmOut ¶ added in v0.8.0

func Atg_SparseAddmmOut(ptr *Ctensor, out Ctensor, self Ctensor, mat1 Ctensor, mat2 Ctensor)

func Atg_SparseBroadcastTo ¶ added in v0.7.0

func Atg_SparseBroadcastTo(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func Atg_SparseBroadcastToCopy ¶ added in v0.8.0

func Atg_SparseBroadcastToCopy(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func Atg_SparseBroadcastToCopyOut ¶ added in v0.8.0

func Atg_SparseBroadcastToCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func Atg_SparseBscTensorUnsafe ¶ added in v0.8.0

func Atg_SparseBscTensorUnsafe(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func Atg_SparseBsrTensorUnsafe ¶ added in v0.8.0

func Atg_SparseBsrTensorUnsafe(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func Atg_SparseCompressedTensorUnsafe ¶ added in v0.8.0

func Atg_SparseCompressedTensorUnsafe(ptr *Ctensor, compressedIndices Ctensor, plainIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func Atg_SparseCooTensorUnsafe ¶

func Atg_SparseCooTensorUnsafe(ptr *Ctensor, indices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32, isCoalesced int32)

func Atg_SparseCooTensorWithDims ¶

func Atg_SparseCooTensorWithDims(ptr *Ctensor, sparseDim int64, denseDim int64, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func Atg_SparseCooTensorWithDimsAndTensors ¶

func Atg_SparseCooTensorWithDimsAndTensors(ptr *Ctensor, sparseDim int64, denseDim int64, sizeData []int64, sizeLen int, indices Ctensor, values Ctensor, optionsKind int32, optionsDevice int32, isCoalesced int32)

func Atg_SparseCooTensorWithDimsAndTensorsOut ¶ added in v0.8.0

func Atg_SparseCooTensorWithDimsAndTensorsOut(ptr *Ctensor, out Ctensor, sparseDim int64, denseDim int64, sizeData []int64, sizeLen int, indices Ctensor, values Ctensor, isCoalesced int32)

func Atg_SparseCooTensorWithDimsOut ¶ added in v0.8.0

func Atg_SparseCooTensorWithDimsOut(ptr *Ctensor, out Ctensor, sparseDim int64, denseDim int64, sizeData []int64, sizeLen int)

func Atg_SparseCscTensorUnsafe ¶ added in v0.8.0

func Atg_SparseCscTensorUnsafe(ptr *Ctensor, ccolIndices Ctensor, rowIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func Atg_SparseCsrProd ¶ added in v0.8.0

func Atg_SparseCsrProd(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func Atg_SparseCsrProdDimDtypeOut ¶ added in v0.8.0

func Atg_SparseCsrProdDimDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func Atg_SparseCsrSum ¶ added in v0.8.0

func Atg_SparseCsrSum(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func Atg_SparseCsrSumDimDtypeOut ¶ added in v0.8.0

func Atg_SparseCsrSumDimDtypeOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, keepdim int32, dtype int32)

func Atg_SparseCsrTensorUnsafe ¶ added in v0.5.0

func Atg_SparseCsrTensorUnsafe(ptr *Ctensor, crowIndices Ctensor, colIndices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice int32)

func Atg_SparseLogSoftmax ¶ added in v0.3.0

func Atg_SparseLogSoftmax(ptr *Ctensor, self Ctensor, dim int64, halfToFloat int32)

func Atg_SparseLogSoftmaxBackwardData ¶ added in v0.3.0

func Atg_SparseLogSoftmaxBackwardData(ptr *Ctensor, gradOutput Ctensor, output Ctensor, dim int64, self Ctensor)

func Atg_SparseLogSoftmaxBackwardDataOut ¶ added in v0.8.0

func Atg_SparseLogSoftmaxBackwardDataOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, dim int64, self Ctensor)

func Atg_SparseLogSoftmaxInt ¶ added in v0.4.0

func Atg_SparseLogSoftmaxInt(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func Atg_SparseLogSoftmaxOut ¶ added in v0.8.0

func Atg_SparseLogSoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, halfToFloat int32)

func Atg_SparseMaskProjection ¶ added in v0.9.0

func Atg_SparseMaskProjection(ptr *Ctensor, self Ctensor, mask Ctensor, accumulateMatches int32)

func Atg_SparseMaskProjectionOut ¶ added in v0.9.0

func Atg_SparseMaskProjectionOut(ptr *Ctensor, out Ctensor, self Ctensor, mask Ctensor, accumulateMatches int32)

func Atg_SparseMm ¶

func Atg_SparseMm(ptr *Ctensor, sparse Ctensor, dense Ctensor)

func Atg_SparseMmReduce ¶ added in v0.8.0

func Atg_SparseMmReduce(ptr *Ctensor, sparse Ctensor, dense Ctensor, reduce string)

func Atg_SparseMmReduceImpl ¶ added in v0.8.0

func Atg_SparseMmReduceImpl(ptr *Ctensor, self Ctensor, other Ctensor, reduce string)

func Atg_SparseSemiStructuredLinear ¶ added in v0.9.0

func Atg_SparseSemiStructuredLinear(ptr *Ctensor, input Ctensor, weight Ctensor, meta Ctensor, bias Ctensor, activation string)

func Atg_SparseSoftmax ¶ added in v0.3.0

func Atg_SparseSoftmax(ptr *Ctensor, self Ctensor, dim int64, halfToFloat int32)

func Atg_SparseSoftmaxBackwardData ¶ added in v0.3.0

func Atg_SparseSoftmaxBackwardData(ptr *Ctensor, gradOutput Ctensor, output Ctensor, dim int64, self Ctensor)

func Atg_SparseSoftmaxBackwardDataOut ¶ added in v0.8.0

func Atg_SparseSoftmaxBackwardDataOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, dim int64, self Ctensor)

func Atg_SparseSoftmaxInt ¶ added in v0.4.0

func Atg_SparseSoftmaxInt(ptr *Ctensor, self Ctensor, dim int64, dtype int32)

func Atg_SparseSoftmaxOut ¶ added in v0.8.0

func Atg_SparseSoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, halfToFloat int32)

func Atg_SparseSparseMatmul ¶ added in v0.4.0

func Atg_SparseSparseMatmul(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg_SparseSparseMatmulOut ¶ added in v0.8.0

func Atg_SparseSparseMatmulOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func Atg_SparseSum ¶

func Atg_SparseSum(ptr *Ctensor, self Ctensor)

func Atg_SparseSumBackward ¶

func Atg_SparseSumBackward(ptr *Ctensor, grad Ctensor, self Ctensor, dimData []int64, dimLen int)

func Atg_SparseSumBackwardOut ¶ added in v0.8.0

func Atg_SparseSumBackwardOut(ptr *Ctensor, out Ctensor, grad Ctensor, self Ctensor, dimData []int64, dimLen int)

func Atg_SparseSumDim ¶ added in v0.4.0

func Atg_SparseSumDim(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int)

func Atg_SparseSumDimDtype ¶ added in v0.4.0

func Atg_SparseSumDimDtype(ptr *Ctensor, self Ctensor, dimData []int64, dimLen int, dtype int32)

func Atg_SparseSumDimOut ¶ added in v0.8.0

func Atg_SparseSumDimOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int)

func Atg_SparseSumDtype ¶ added in v0.4.0

func Atg_SparseSumDtype(ptr *Ctensor, self Ctensor, dtype int32)

func Atg_Spdiags ¶ added in v0.8.0

func Atg_Spdiags(ptr *Ctensor, diagonals Ctensor, offsets Ctensor, shapeData []int64, shapeLen int, layout int8)

func Atg_SpdiagsOut ¶ added in v0.8.0

func Atg_SpdiagsOut(ptr *Ctensor, out Ctensor, diagonals Ctensor, offsets Ctensor, shapeData []int64, shapeLen int, layout int8)

func Atg_Stack ¶ added in v0.4.0

func Atg_Stack(ptr *Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func Atg_StackOut ¶ added in v0.4.0

func Atg_StackOut(ptr *Ctensor, out Ctensor, tensorsData []Ctensor, tensorsLen int, dim int64)

func Atg_StandardGamma ¶

func Atg_StandardGamma(ptr *Ctensor, self Ctensor)

func Atg_StandardGammaGrad ¶

func Atg_StandardGammaGrad(ptr *Ctensor, self Ctensor, output Ctensor)

func Atg_StandardGammaGradOut ¶ added in v0.8.0

func Atg_StandardGammaGradOut(ptr *Ctensor, out Ctensor, self Ctensor, output Ctensor)

func Atg_StandardGammaOut ¶ added in v0.8.0

func Atg_StandardGammaOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_TestAmbiguousDefaults ¶ added in v0.4.0

func Atg_TestAmbiguousDefaults(ptr *Ctensor, dummy Ctensor, a int64, b int64)

func Atg_TestAmbiguousDefaultsB ¶ added in v0.4.0

func Atg_TestAmbiguousDefaultsB(ptr *Ctensor, dummy Ctensor, a int64, b string)

func Atg_TestAutogradMultipleDispatch ¶ added in v0.8.0

func Atg_TestAutogradMultipleDispatch(ptr *Ctensor, self Ctensor)

func Atg_TestAutogradMultipleDispatchFullcoverageOut ¶ added in v0.8.0

func Atg_TestAutogradMultipleDispatchFullcoverageOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_TestAutogradMultipleDispatchNtonly ¶ added in v0.8.0

func Atg_TestAutogradMultipleDispatchNtonly(ptr *Ctensor, self Ctensor, b int32)

func Atg_TestAutogradMultipleDispatchView ¶ added in v0.8.0

func Atg_TestAutogradMultipleDispatchView(ptr *Ctensor, self Ctensor)

func Atg_TestAutogradMultipleDispatchViewCopy ¶ added in v0.8.0

func Atg_TestAutogradMultipleDispatchViewCopy(ptr *Ctensor, self Ctensor)

func Atg_TestAutogradMultipleDispatchViewCopyOut ¶ added in v0.8.0

func Atg_TestAutogradMultipleDispatchViewCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_TestCheckTensor ¶ added in v0.8.0

func Atg_TestCheckTensor(ptr *Ctensor, self Ctensor)

func Atg_TestFunctorchFallback ¶ added in v0.9.0

func Atg_TestFunctorchFallback(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg_TestFunctorchFallbackOut ¶ added in v0.9.0

func Atg_TestFunctorchFallbackOut(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func Atg_TestOptionalFilledIntlist ¶ added in v0.3.0

func Atg_TestOptionalFilledIntlist(ptr *Ctensor, values Ctensor, addendsData []int64, addendsLen int)

func Atg_TestOptionalFilledIntlistOut ¶ added in v0.8.0

func Atg_TestOptionalFilledIntlistOut(ptr *Ctensor, out Ctensor, values Ctensor, addendsData []int64, addendsLen int)

func Atg_TestOptionalFloatlist ¶ added in v0.8.0

func Atg_TestOptionalFloatlist(ptr *Ctensor, values Ctensor, addendsData []float64, addendsLen int)

func Atg_TestOptionalFloatlistOut ¶ added in v0.8.0

func Atg_TestOptionalFloatlistOut(ptr *Ctensor, out Ctensor, values Ctensor, addendsData []float64, addendsLen int)

func Atg_TestOptionalIntlist ¶ added in v0.3.0

func Atg_TestOptionalIntlist(ptr *Ctensor, values Ctensor, addendsData []int64, addendsLen int)

func Atg_TestOptionalIntlistOut ¶ added in v0.8.0

func Atg_TestOptionalIntlistOut(ptr *Ctensor, out Ctensor, values Ctensor, addendsData []int64, addendsLen int)

func Atg_TestSerializationSubcmul ¶ added in v0.3.0

func Atg_TestSerializationSubcmul(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg_TestStringDefault ¶ added in v0.4.0

func Atg_TestStringDefault(ptr *Ctensor, dummy Ctensor, a string, b string)

func Atg_TestWarnInAutograd ¶ added in v0.7.0

func Atg_TestWarnInAutograd(ptr *Ctensor, self Ctensor)

func Atg_TestWarnInAutogradOut ¶ added in v0.8.0

func Atg_TestWarnInAutogradOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_ToCopy ¶ added in v0.5.0

func Atg_ToCopy(ptr *Ctensor, self Ctensor, optionsKind int32, optionsDevice int32, nonBlocking int32)

func Atg_ToCopyOut ¶ added in v0.8.0

func Atg_ToCopyOut(ptr *Ctensor, out Ctensor, self Ctensor, nonBlocking int32)

func Atg_ToDense ¶ added in v0.8.0

func Atg_ToDense(ptr *Ctensor, self Ctensor, dtype int32, maskedGrad int32)

func Atg_ToDenseOut ¶ added in v0.8.0

func Atg_ToDenseOut(ptr *Ctensor, out Ctensor, self Ctensor, dtype int32, maskedGrad int32)

func Atg_ToSparse ¶ added in v0.9.0

func Atg_ToSparse(ptr *Ctensor, self Ctensor, layout int8, blocksizeData []int64, blocksizeLen int, denseDimVal int64, denseDimNull int)

func Atg_ToSparseBsc ¶ added in v0.9.0

func Atg_ToSparseBsc(ptr *Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, denseDimVal int64, denseDimNull int)

func Atg_ToSparseBscOut ¶ added in v0.9.0

func Atg_ToSparseBscOut(ptr *Ctensor, out Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, denseDimVal int64, denseDimNull int)

func Atg_ToSparseBsr ¶ added in v0.9.0

func Atg_ToSparseBsr(ptr *Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, denseDimVal int64, denseDimNull int)

func Atg_ToSparseBsrOut ¶ added in v0.9.0

func Atg_ToSparseBsrOut(ptr *Ctensor, out Ctensor, self Ctensor, blocksizeData []int64, blocksizeLen int, denseDimVal int64, denseDimNull int)

func Atg_ToSparseCsc ¶ added in v0.9.0

func Atg_ToSparseCsc(ptr *Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)

func Atg_ToSparseCscOut ¶ added in v0.9.0

func Atg_ToSparseCscOut(ptr *Ctensor, out Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)

func Atg_ToSparseCsr ¶ added in v0.9.0

func Atg_ToSparseCsr(ptr *Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)

func Atg_ToSparseCsrOut ¶ added in v0.9.0

func Atg_ToSparseCsrOut(ptr *Ctensor, out Ctensor, self Ctensor, denseDimVal int64, denseDimNull int)

func Atg_ToSparseOut ¶ added in v0.9.0

func Atg_ToSparseOut(ptr *Ctensor, out Ctensor, self Ctensor, layout int8, blocksizeData []int64, blocksizeLen int, denseDimVal int64, denseDimNull int)

func Atg_ToSparseSemiStructured ¶ added in v0.9.0

func Atg_ToSparseSemiStructured(ptr *Ctensor, dense Ctensor)

func Atg_ToSparseSparseDim ¶ added in v0.9.0

func Atg_ToSparseSparseDim(ptr *Ctensor, self Ctensor, sparseDim int64)

func Atg_ToSparseSparseDimOut ¶ added in v0.9.0

func Atg_ToSparseSparseDimOut(ptr *Ctensor, out Ctensor, self Ctensor, sparseDim int64)

func Atg_TransformBiasRescaleQkv ¶ added in v0.8.0

func Atg_TransformBiasRescaleQkv(ptr *Ctensor, qkv Ctensor, qkvBias Ctensor, numHeads int64)

func Atg_TransformBiasRescaleQkvOut ¶ added in v0.8.0

func Atg_TransformBiasRescaleQkvOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, qkv Ctensor, qkvBias Ctensor, numHeads int64)

func Atg_TransformerEncoderLayerFwd ¶ added in v0.8.0

func Atg_TransformerEncoderLayerFwd(ptr *Ctensor, src Ctensor, embedDim int64, numHeads int64, qkvWeight Ctensor, qkvBias Ctensor, projWeight Ctensor, projBias Ctensor, useGelu int32, normFirst int32, eps float64, normWeight1 Ctensor, normBias1 Ctensor, normWeight2 Ctensor, normBias2 Ctensor, ffnWeight1 Ctensor, ffnBias1 Ctensor, ffnWeight2 Ctensor, ffnBias2 Ctensor, mask Ctensor, maskTypeVal int64, maskTypeNull int)

func Atg_TransformerEncoderLayerFwdOut ¶ added in v0.8.0

func Atg_TransformerEncoderLayerFwdOut(ptr *Ctensor, out Ctensor, src Ctensor, embedDim int64, numHeads int64, qkvWeight Ctensor, qkvBias Ctensor, projWeight Ctensor, projBias Ctensor, useGelu int32, normFirst int32, eps float64, normWeight1 Ctensor, normBias1 Ctensor, normWeight2 Ctensor, normBias2 Ctensor, ffnWeight1 Ctensor, ffnBias1 Ctensor, ffnWeight2 Ctensor, ffnBias2 Ctensor, mask Ctensor, maskTypeVal int64, maskTypeNull int)

func Atg_Trilinear ¶

func Atg_Trilinear(ptr *Ctensor, i1 Ctensor, i2 Ctensor, i3 Ctensor, expand1Data []int64, expand1Len int, expand2Data []int64, expand2Len int, expand3Data []int64, expand3Len int, sumdimData []int64, sumdimLen int, unrollDim int64)

func Atg_TrilinearOut ¶ added in v0.8.0

func Atg_TrilinearOut(ptr *Ctensor, out Ctensor, i1 Ctensor, i2 Ctensor, i3 Ctensor, expand1Data []int64, expand1Len int, expand2Data []int64, expand2Len int, expand3Data []int64, expand3Len int, sumdimData []int64, sumdimLen int, unrollDim int64)

func Atg_TritonMultiHeadAttention ¶ added in v0.8.0

func Atg_TritonMultiHeadAttention(ptr *Ctensor, query Ctensor, key Ctensor, value Ctensor, embedDim int64, numHead int64, qkvWeight Ctensor, qkvBias Ctensor, projWeight Ctensor, projBias Ctensor, mask Ctensor)

func Atg_TritonMultiHeadAttentionOut ¶ added in v0.8.0

func Atg_TritonMultiHeadAttentionOut(ptr *Ctensor, out Ctensor, query Ctensor, key Ctensor, value Ctensor, embedDim int64, numHead int64, qkvWeight Ctensor, qkvBias Ctensor, projWeight Ctensor, projBias Ctensor, mask Ctensor)

func Atg_TritonScaledDotAttention ¶ added in v0.8.0

func Atg_TritonScaledDotAttention(ptr *Ctensor, q Ctensor, k Ctensor, v Ctensor, dropoutP float64)

func Atg_TritonScaledDotAttentionOut ¶ added in v0.8.0

func Atg_TritonScaledDotAttentionOut(ptr *Ctensor, out Ctensor, q Ctensor, k Ctensor, v Ctensor, dropoutP float64)

func Atg_Unique ¶

func Atg_Unique(ptr *Ctensor, self Ctensor, sorted int32, returnInverse int32)

func Atg_Unique2 ¶

func Atg_Unique2(ptr *Ctensor, self Ctensor, sorted int32, returnInverse int32, returnCounts int32)

func Atg_Unique2Out ¶ added in v0.8.0

func Atg_Unique2Out(ptr *Ctensor, out0 Ctensor, out1 Ctensor, out2 Ctensor, self Ctensor, sorted int32, returnInverse int32, returnCounts int32)

func Atg_UniqueOut ¶ added in v0.8.0

func Atg_UniqueOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, self Ctensor, sorted int32, returnInverse int32)

func Atg_UnpackDual ¶ added in v0.4.0

func Atg_UnpackDual(ptr *Ctensor, dual Ctensor, level int64)

func Atg_UnsafeIndex ¶ added in v0.9.0

func Atg_UnsafeIndex(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int)

func Atg_UnsafeIndexPut ¶ added in v0.9.0

func Atg_UnsafeIndexPut(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate int32)

func Atg_UnsafeView ¶

func Atg_UnsafeView(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func Atg_UnsafeViewOut ¶ added in v0.8.0

func Atg_UnsafeViewOut(ptr *Ctensor, out Ctensor, self Ctensor, sizeData []int64, sizeLen int)

func Atg_UpsampleBicubic2dAa ¶ added in v0.7.0

func Atg_UpsampleBicubic2dAa(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleBicubic2dAaBackward ¶ added in v0.7.0

func Atg_UpsampleBicubic2dAaBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleBicubic2dAaBackwardGradInput ¶ added in v0.7.0

func Atg_UpsampleBicubic2dAaBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleBicubic2dAaOut ¶ added in v0.7.0

func Atg_UpsampleBicubic2dAaOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleBicubic2dAaVec ¶ added in v0.8.0

func Atg_UpsampleBicubic2dAaVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scaleFactorsData []float64, scaleFactorsLen int)

func Atg_UpsampleBilinear2dAa ¶ added in v0.7.0

func Atg_UpsampleBilinear2dAa(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleBilinear2dAaBackward ¶ added in v0.7.0

func Atg_UpsampleBilinear2dAaBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleBilinear2dAaBackwardGradInput ¶ added in v0.7.0

func Atg_UpsampleBilinear2dAaBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleBilinear2dAaOut ¶ added in v0.7.0

func Atg_UpsampleBilinear2dAaOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleBilinear2dAaVec ¶ added in v0.8.0

func Atg_UpsampleBilinear2dAaVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, alignCorners int32, scaleFactorsData []float64, scaleFactorsLen int)

func Atg_UpsampleNearestExact1d ¶ added in v0.7.0

func Atg_UpsampleNearestExact1d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesVal float64, scalesNull int)

func Atg_UpsampleNearestExact1dBackward ¶ added in v0.7.0

func Atg_UpsampleNearestExact1dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesVal float64, scalesNull int)

func Atg_UpsampleNearestExact1dBackwardGradInput ¶ added in v0.7.0

func Atg_UpsampleNearestExact1dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesVal float64, scalesNull int)

func Atg_UpsampleNearestExact1dOut ¶ added in v0.7.0

func Atg_UpsampleNearestExact1dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesVal float64, scalesNull int)

func Atg_UpsampleNearestExact1dVec ¶ added in v0.8.0

func Atg_UpsampleNearestExact1dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, scaleFactorsData []float64, scaleFactorsLen int)

func Atg_UpsampleNearestExact2d ¶ added in v0.7.0

func Atg_UpsampleNearestExact2d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleNearestExact2dBackward ¶ added in v0.7.0

func Atg_UpsampleNearestExact2dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleNearestExact2dBackwardGradInput ¶ added in v0.7.0

func Atg_UpsampleNearestExact2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleNearestExact2dOut ¶ added in v0.7.0

func Atg_UpsampleNearestExact2dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleNearestExact2dVec ¶ added in v0.8.0

func Atg_UpsampleNearestExact2dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, scaleFactorsData []float64, scaleFactorsLen int)

func Atg_UpsampleNearestExact3d ¶ added in v0.7.0

func Atg_UpsampleNearestExact3d(ptr *Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleNearestExact3dBackward ¶ added in v0.7.0

func Atg_UpsampleNearestExact3dBackward(ptr *Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleNearestExact3dBackwardGradInput ¶ added in v0.7.0

func Atg_UpsampleNearestExact3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, outputSizeData []int64, outputSizeLen int, inputSizeData []int64, inputSizeLen int, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleNearestExact3dOut ¶ added in v0.7.0

func Atg_UpsampleNearestExact3dOut(ptr *Ctensor, out Ctensor, self Ctensor, outputSizeData []int64, outputSizeLen int, scalesDVal float64, scalesDNull int, scalesHVal float64, scalesHNull int, scalesWVal float64, scalesWNull int)

func Atg_UpsampleNearestExact3dVec ¶ added in v0.8.0

func Atg_UpsampleNearestExact3dVec(ptr *Ctensor, input Ctensor, outputSizeData []int64, outputSizeLen int, scaleFactorsData []float64, scaleFactorsLen int)

func Atg_UseCudnnCtcLoss ¶ added in v0.4.0

func Atg_UseCudnnCtcLoss(logProbs Ctensor, targets Ctensor, inputLengthsData []int64, inputLengthsLen int, targetLengthsData []int64, targetLengthsLen int, blank int64) bool

func Atg_UseCudnnCtcLossTensor ¶ added in v0.8.0

func Atg_UseCudnnCtcLossTensor(logProbs Ctensor, targets Ctensor, inputLengths Ctensor, targetLengths Ctensor, blank int64) bool

func Atg_UseCudnnRnnFlattenWeight ¶ added in v0.4.0

func Atg_UseCudnnRnnFlattenWeight() bool

func Atg_Values ¶

func Atg_Values(ptr *Ctensor, self Ctensor)

func Atg_ValuesCopy ¶ added in v0.8.0

func Atg_ValuesCopy(ptr *Ctensor, self Ctensor)

func Atg_ValuesCopyOut ¶ added in v0.8.0

func Atg_ValuesCopyOut(ptr *Ctensor, out Ctensor, self Ctensor)

func Atg_Version ¶ added in v0.4.0

func Atg_Version(self Ctensor) int64

func Atg_WeightNorm ¶

func Atg_WeightNorm(ptr *Ctensor, v Ctensor, g Ctensor, dim int64)

func Atg_WeightNormDifferentiableBackward ¶

func Atg_WeightNormDifferentiableBackward(ptr *Ctensor, gradW Ctensor, savedV Ctensor, savedG Ctensor, savedNorms Ctensor, dim int64)

func Atg_WeightNormInterface ¶ added in v0.8.0

func Atg_WeightNormInterface(ptr *Ctensor, v Ctensor, g Ctensor, dim int64)

func Atg_WeightNormInterfaceBackward ¶ added in v0.8.0

func Atg_WeightNormInterfaceBackward(ptr *Ctensor, gradW Ctensor, savedV Ctensor, savedG Ctensor, savedNorms Ctensor, dim int64)

func Atg_WeightNormInterfaceBackwardOut ¶ added in v0.8.0

func Atg_WeightNormInterfaceBackwardOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, gradW Ctensor, savedV Ctensor, savedG Ctensor, savedNorms Ctensor, dim int64)

func Atg_WeightNormInterfaceOut ¶ added in v0.8.0

func Atg_WeightNormInterfaceOut(ptr *Ctensor, out0 Ctensor, out1 Ctensor, v Ctensor, g Ctensor, dim int64)

func Atg__AndTensor_ ¶ added in v0.4.0

func Atg__AndTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__And_ ¶

func Atg__And_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg__IandTensor_ ¶ added in v0.4.0

func Atg__IandTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__Iand_ ¶

func Atg__Iand_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg__IlshiftTensor_ ¶ added in v0.4.0

func Atg__IlshiftTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__Ilshift_ ¶

func Atg__Ilshift_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg__IorTensor_ ¶ added in v0.4.0

func Atg__IorTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__Ior_ ¶

func Atg__Ior_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg__IrshiftTensor_ ¶ added in v0.4.0

func Atg__IrshiftTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__Irshift_ ¶

func Atg__Irshift_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg__IxorTensor_ ¶ added in v0.4.0

func Atg__IxorTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__Ixor_ ¶

func Atg__Ixor_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg__LshiftScalarOut_ ¶ added in v0.8.0

func Atg__LshiftScalarOut_(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func Atg__LshiftTensorOut_ ¶ added in v0.8.0

func Atg__LshiftTensorOut_(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func Atg__LshiftTensor_ ¶ added in v0.4.0

func Atg__LshiftTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__Lshift_ ¶

func Atg__Lshift_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg__OrTensor_ ¶ added in v0.4.0

func Atg__OrTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__Or_ ¶

func Atg__Or_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg__RshiftScalarOut_ ¶ added in v0.8.0

func Atg__RshiftScalarOut_(ptr *Ctensor, out Ctensor, self Ctensor, other Cscalar)

func Atg__RshiftTensorOut_ ¶ added in v0.8.0

func Atg__RshiftTensorOut_(ptr *Ctensor, out Ctensor, self Ctensor, other Ctensor)

func Atg__RshiftTensor_ ¶ added in v0.4.0

func Atg__RshiftTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__Rshift_ ¶

func Atg__Rshift_(ptr *Ctensor, self Ctensor, other Cscalar)

func Atg__XorTensor_ ¶ added in v0.4.0

func Atg__XorTensor_(ptr *Ctensor, self Ctensor, other Ctensor)

func Atg__Xor_ ¶

func Atg__Xor_(ptr *Ctensor, self Ctensor, other Cscalar)

func AtiFree ¶

func AtiFree(val Civalue)

void ati_free(ivalue);

func AtiLength ¶

func AtiLength(val Civalue) int32

int ati_length(ivalue);

func AtiTag ¶

func AtiTag(val Civalue) int32

int ati_tag(ivalue);

func AtiToBool ¶

func AtiToBool(val Civalue) bool

int ati_to_bool(ivalue);

func AtiToBoolList ¶

func AtiToBoolList(val Civalue, ptr unsafe.Pointer, boolListLen int)

void ati_to_bool_list(ivalue, char *, int);

func AtiToDouble ¶

func AtiToDouble(val Civalue) float64

double ati_to_double(ivalue);

func AtiToDoubleList ¶

func AtiToDoubleList(val Civalue, ptr unsafe.Pointer, doubleListLen int)

void ati_to_double_list(ivalue, double *, int);

func AtiToGenericDict ¶

func AtiToGenericDict(val Civalue, ptr *Civalue, genericDictLen int)

void ati_to_generic_dict(ivalue, ivalue *, int);

func AtiToGenericList ¶

func AtiToGenericList(val Civalue, ptr *Civalue, genericListLen int)

void ati_to_generic_list(ivalue, ivalue *, int);

func AtiToInt ¶

func AtiToInt(val Civalue) int64

int64_t ati_to_int(ivalue);

func AtiToIntList ¶

func AtiToIntList(val Civalue, ptr unsafe.Pointer, intListLen int)

void ati_to_int_list(ivalue, int64_t *, int);

func AtiToString ¶

func AtiToString(val Civalue) string

char *ati_to_string(ivalue);

func AtiToTensorList ¶

func AtiToTensorList(val Civalue, ptr *Ctensor, tensorListLen int)

void ati_to_tensor_list(ivalue, tensor *, int);

func AtiToTuple ¶

func AtiToTuple(val Civalue, ptr *Civalue, tupleLen int)

void ati_to_tuple(ivalue, ivalue *, int);

func AtiTupleLength ¶

func AtiTupleLength(val Civalue) int32

int ati_tuple_length(ivalue);

func AtmEval ¶ added in v0.3.7

func AtmEval(m Cmodule)

void atm_eval(module);

func AtmFree ¶

func AtmFree(m Cmodule)

void atm_free(module);

func AtmGetProfilingMode ¶ added in v0.3.7

func AtmGetProfilingMode() bool

int atm_get_profiling_mode();

func AtmNamedParameters ¶ added in v0.3.7

func AtmNamedParameters(m Cmodule, dataPtr unsafe.Pointer)

void atm_named_parameters(module, void *data, void (*f)(void *, char *, tensor));

func AtmSave ¶ added in v0.3.7

func AtmSave(m Cmodule, path string)

void atm_save(module m, char *);

func AtmSetProfilingMode ¶ added in v0.3.7

func AtmSetProfilingMode(b bool)

void atm_set_profiling_mode(int);

func AtmTo ¶

func AtmTo(m Cmodule, device int32, dtype int32, nonBlocking bool)

void atm_to(module m, int device, int dtype, bool non_blocking);

func AtmTrain ¶ added in v0.3.7

func AtmTrain(m Cmodule)

void atm_train(module);

func AtoAddParamGroup ¶ added in v0.3.10

func AtoAddParamGroup(coptimizer Coptimizer, tensors []Ctensor, ntensors int)

func AtoAddParameter ¶ added in v0.3.10

func AtoAddParameter(coptimizer Coptimizer, tensor Ctensor, group uint)

func AtoAddParametersOld ¶ added in v0.3.0

func AtoAddParametersOld(coptimizer Coptimizer, tensors []Ctensor, ntensors int)

NOTE. Backward compat for param group not updated (#261) void ato_add_parameters(optimizer, tensor *, int ntensors);

func AtoConstantPadNd ¶ added in v0.6.0

func AtoConstantPadNd(ptr *Ctensor, self Ctensor, padData []int64, padLen int, value Cscalar)

func AtoFree ¶

func AtoFree(coptimizer Coptimizer)

void ato_free(optimizer);

func AtoGetLearningRates ¶ added in v0.3.10

func AtoGetLearningRates(coptimizer Coptimizer) []float64

func AtoParamGroupNum ¶ added in v0.3.10

func AtoParamGroupNum(coptimizer Coptimizer) int64

func AtoSetLearningRate ¶

func AtoSetLearningRate(coptimizer Coptimizer, learningRate float64)

void ato_set_learning_rate(optimizer, double learning_rate);

func AtoSetLearningRates ¶ added in v0.3.10

func AtoSetLearningRates(coptimizer Coptimizer, lrs []float64)

func AtoSetMomentum ¶

func AtoSetMomentum(coptimizer Coptimizer, momentum float64)

void ato_set_momentum(optimizer, double momentum);

func AtoStep ¶

func AtoStep(coptimizer Coptimizer)

void ato_step(optimizer);

func AtoZeroGrad ¶

func AtoZeroGrad(coptimizer Coptimizer)

void ato_zero_grad(optimizer);

func AtsFree ¶

func AtsFree(cscalar Cscalar)

void ats_free(scalar);

func AtsToFloat ¶

func AtsToFloat(cscalar Cscalar) float64

double ats_to_float(scalar);

func AtsToInt ¶

func AtsToInt(cscalar Cscalar) int64

int64_t ats_to_int(scalar);

func AtsToString ¶

func AtsToString(cscalar Cscalar) string

char *ats_to_string(scalar);

func GetAndResetLastErr ¶

func GetAndResetLastErr() *C.char

func IsNull ¶ added in v0.8.0

func IsNull(ctensor Ctensor) bool

Types ¶

type Civalue ¶

type Civalue = C.ivalue

func AtiBool ¶

func AtiBool(val bool) Civalue

ivalue ati_bool(int);

func AtiBoolList ¶

func AtiBoolList(boolListData []bool, boolListLen int) Civalue

ivalue ati_bool_list(char *, int);

func AtiDouble ¶

func AtiDouble(val float64) Civalue

ivalue ati_double(double);

func AtiDoubleList ¶

func AtiDoubleList(doubleListData []float64, doubleListLen int) Civalue

ivalue ati_double_list(double *, int);

func AtiGenericDict ¶

func AtiGenericDict(genericDictData []Civalue, genericDictLen int) Civalue

ivalue ati_generic_dict(ivalue *, int);

func AtiGenericList ¶

func AtiGenericList(genericListData []Civalue, genericListLen int) Civalue

ivalue ati_generic_list(ivalue *, int);

func AtiInt ¶

func AtiInt(val int64) Civalue

ivalue ati_int(int64_t);

func AtiIntList ¶

func AtiIntList(intListData []int64, intListLen int) Civalue

ivalue ati_int_list(int64_t *, int);

func AtiNone ¶

func AtiNone() Civalue

ivalue ati_none();

func AtiString ¶

func AtiString(val string) Civalue

ivalue ati_string(char *);

func AtiTensor ¶

func AtiTensor(ts Ctensor) Civalue

ivalue ati_tensor(tensor);

func AtiTensorList ¶

func AtiTensorList(tensorListData []Ctensor, tensorListLen int) Civalue

ivalue ati_tensor_list(tensor *, int);

func AtiTuple ¶

func AtiTuple(tupleData []Civalue, tupleLen int) Civalue

ivalue ati_tuple(ivalue *, int);

func AtmForward_ ¶

func AtmForward_(m Cmodule, ivalues *Civalue, nivalues int) Civalue

ivalue atm_forward_(module, ivalue *ivalues, int nivalues);

type Cmodule ¶

type Cmodule = C.module

func AtmLoad ¶

func AtmLoad(path string) Cmodule

module atm_load(char *);

func AtmLoadOnDevice ¶

func AtmLoadOnDevice(path string, device int32) Cmodule

module atm_load_on_device(char *, int device);

func AtmLoadStr ¶

func AtmLoadStr(val string, sz int) Cmodule

module atm_load_str(char *, size_t sz);

func AtmLoadStrOnDevice ¶

func AtmLoadStrOnDevice(val string, sz int, device int32) Cmodule

module atm_load_str_on_device(char *, size_t sz, int device);

type Coptimizer ¶

type Coptimizer = C.optimizer

func AtoAdam ¶

func AtoAdam(learningRate, beta1, beta2, weightDecay float64) Coptimizer

* optimizer ato_adam(double learning_rate, * double beta1, * double beta2, * double weight_decay); *

func AtoAdamW ¶ added in v0.3.11

func AtoAdamW(learningRate, beta1, beta2, weightDecay float64) Coptimizer

func AtoRmsProp ¶

func AtoRmsProp(learningRate, alpha, eps, weightDecay, momentum float64, centered int) Coptimizer

* optimizer ato_rms_prop(double learning_rate, * double alpha, * double eps, * double weight_decay, * double momentum, * int centered); *

func AtoSgd ¶

func AtoSgd(learningRate, momentum, dampening, weightDecay float64, nesterov int) Coptimizer

* optimizer ato_sgd(double learning_rate, * double momentum, * double dampening, * double weight_decay, * int nesterov); *

type Cscalar ¶

type Cscalar = C.scalar

func AtsFloat ¶

func AtsFloat(v float64) Cscalar

scalar ats_float(double);

func AtsInt ¶

func AtsInt(v int64) Cscalar

scalar ats_int(int64_t);

type Ctensor ¶

type Ctensor = C.tensor

NOTE: C.tensor is a C pointer to torch::Tensor

func AtGet ¶

func AtGet(ts Ctensor, index int) Ctensor

tensor at_get(tensor, int index);

func AtLoad ¶

func AtLoad(path string) Ctensor

tensor at_load(char *filename);

func AtLoadImage ¶

func AtLoadImage(path string) Ctensor

tensor at_load_image(char *filename);

func AtNewTensor ¶

func AtNewTensor() Ctensor

func AtResizeImage ¶

func AtResizeImage(ts Ctensor, w, h int64) Ctensor

tensor at_resize_image(tensor, int w, int h);

func AtShallowClone ¶

func AtShallowClone(ts Ctensor) Ctensor

tensor at_shallow_clone(tensor);

func AtTensorOfData ¶

func AtTensorOfData(vs unsafe.Pointer, dims []int64, ndims uint, elt_size_in_bytes uint, kind int) Ctensor

tensor at_tensor_of_data(void *vs, int64_t *dims, size_t ndims, size_t element_size_in_bytes, int type);

func AtgAlignTensors ¶

func AtgAlignTensors(tensorsData []Ctensor, tensorsLen int) *Ctensor

tensor *atg_align_tensors(tensor *tensors_data, int tensors_len);

func AtgBroadcastTensors ¶

func AtgBroadcastTensors(tensorsData []Ctensor, tensorsLen int) *Ctensor

tensor *atg_broadcast_tensors(tensor *tensors_data, int tensors_len);

func AtgChunk ¶

func AtgChunk(self Ctensor, chunks int64, dim int64) *Ctensor

tensor *atg_chunk(tensor self, int64_t chunks, int64_t dim);

func AtgMeshgrid ¶

func AtgMeshgrid(tensorsData []Ctensor, tensorsLen int) *Ctensor

tensor *atg_meshgrid(tensor *tensors_data, int tensors_len);

func AtgNonzeroNumpy ¶

func AtgNonzeroNumpy(self Ctensor) *Ctensor

tensor *atg_nonzero_numpy(tensor self);

func AtgSplit ¶

func AtgSplit(self Ctensor, splitSize int64, dim int64) *Ctensor

tensor *atg_split(tensor self, int64_t split_size, int64_t dim);

func AtgSplitWithSizes ¶

func AtgSplitWithSizes(self Ctensor, splitSizesData []int64, splitSizesLen int, dim int64) *Ctensor

tensor *atg_split_with_sizes(tensor self, int64_t *split_sizes_data, int split_sizes_len, int64_t dim);

func AtgUnbind ¶

func AtgUnbind(self Ctensor, dim int64) *Ctensor

tensor *atg_unbind(tensor self, int64_t dim);

func AtgWhere ¶

func AtgWhere(condition Ctensor) *Ctensor

tensor *atg_where(tensor condition);

func AtiToTensor ¶

func AtiToTensor(val Civalue) Ctensor

tensor ati_to_tensor(ivalue);

func AtmForward ¶

func AtmForward(m Cmodule, tensors *Ctensor, ntensors int) Ctensor

tensor atm_forward(module, tensor *tensors, int ntensors);

func NewTensor ¶

func NewTensor() Ctensor

tensor at_new_tensor();

type LoadData ¶

type LoadData struct {
	NamedCtensors []NamedCtensor
}

type NamedCtensor ¶

type NamedCtensor struct {
	Name    string
	Ctensor C.tensor
}

type PointerStore ¶

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

PointerStore is a Go struct to deal with the Go use case that we can not pass Go pointer to C. In other words, it is used to solve error: "panic: runtime error: cgo argument has Go pointer to Go pointer"

NOTE: the concept is taken from: https://github.com/mattn/go-pointer

Example:

store := NewPointerStore()
type Car struct{Name string, Model string}
var landy Car{Name: "Defender", Model: "99"}
landyPtr := store.Set(landy)
landy = store.Get(landyPtr).(Car)
store.Free(landyPtr)

func NewPointerStore ¶

func NewPointerStore() PointerStore

NewPointerStore creates a new PointerStore

func (PointerStore) Free ¶

func (ps PointerStore) Free(ptr unsafe.Pointer)

Delete removes pointer from pointer store and frees up memory.

Example: TODO: an example

func (PointerStore) Get ¶

func (ps PointerStore) Get(ptr unsafe.Pointer) (v interface{})

Get get value back from pointer store

Example: TODO: an example

func (PointerStore) Set ¶

func (ps PointerStore) Set(v interface{}) unsafe.Pointer

Set stores value to pointer store and returns a unsafe.Pointer

NOTE: This is a little hacky. As Go doesn't allow C code to store Go pointers, a one-byte C pointer is created for indexing purpose.

Example: TODO: an example

Source Files ¶

View all Source files

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

README ¶

NOTES ON WRITING WRAPPER FUNCTIONS

Function Input Arguments

tensor -> t *C_tensor

C pointer e.g int64_t * -> ptr unsafe.Pointer

C types e.g size_t ndims -> equivalent Go types ndims uint

Function Return

void *CFUNC(...)

tensor -> *C_tensor

tensor *CFUNC(...)

C types e.g. C_ulong -> Go equivalent types uint64

C type pointers e.g. char *FUNCTION() --> *C.char

Multiple Tensors Created In C Land Memory

Documentation ¶

Overview ¶

Index ¶

Constants ¶

Variables ¶

Functions ¶

func AtBackward ¶

func AtCopyData ¶

func AtCopy_ ¶

func AtDataPtr ¶

func AtDefined ¶

func AtDevice ¶

func AtDim ¶

func AtDoubleValueAtIndexes ¶

func AtFree ¶

func AtGradSetEnabled ¶

func AtInt64ValueAtIndexes ¶

func AtIsContiguous ¶ added in v0.8.0

func AtIsMkldnn ¶ added in v0.8.0

func AtIsSparse ¶

func AtLoadCallback ¶

func AtLoadCallbackWithDevice ¶

func AtLoadMulti ¶

func AtPrint ¶

func AtRequiresGrad ¶

func AtRunBackward ¶

func AtSave ¶

func AtSaveImage ¶

func AtSaveMulti ¶

func AtSaveMultiNew ¶ added in v0.1.10

func AtScalarType ¶

func AtShape ¶

func AtStride ¶ added in v0.8.0

func AtToString ¶

func AtcCudaDeviceCount ¶

func AtcCudaIsAvailable ¶

func AtcCudnnIsAvailable ¶

func AtcGetDevice ¶ added in v0.8.0

func AtcSetBenchmarkCudnn ¶

func AtcSetDevice ¶ added in v0.8.0

func AtcSynchronize ¶ added in v0.8.0

func Atc_cuda_device_count ¶

func AtgAbs ¶

func AtgAbsOut ¶

func AtgAbs_ ¶

func AtgAbsolute ¶ added in v0.3.0

func AtgAbsoluteOut ¶ added in v0.3.0

func AtgAbsolute_ ¶ added in v0.3.0

func AtgAcos ¶

func AtgAcosOut ¶

func AtgAcos_ ¶

func AtgAcosh ¶ added in v0.3.0

func AtgAcoshOut ¶ added in v0.3.0

func AtgAcosh_ ¶ added in v0.3.0

func AtgAdaptiveAvgPool1d ¶

func AtgAdaptiveAvgPool2d ¶

func AtgAdaptiveAvgPool2dOut ¶

func AtgAdaptiveAvgPool3d ¶

func AtgAdaptiveAvgPool3dBackward ¶

func AtgAdaptiveAvgPool3dOut ¶

func AtgAdaptiveMaxPool1d ¶

func AtgAdaptiveMaxPool2d ¶

func AtgAdaptiveMaxPool2dBackward ¶

func AtgAdaptiveMaxPool2dBackwardGradInput ¶ added in v0.4.0

func AtgAdaptiveMaxPool2dOut ¶

func AtgAdaptiveMaxPool3d ¶

func AtgAdaptiveMaxPool3dBackward ¶

`tensor` -> `t *C_tensor`

C pointer e.g `int64_t *` -> `ptr unsafe.Pointer`

C types e.g `size_t ndims` -> equivalent Go types `ndims uint`

`void *CFUNC(...)`

`tensor` -> `*C_tensor`

`tensor *CFUNC(...)`

C types e.g. `C_ulong` -> Go equivalent types `uint64`

C type pointers e.g. `char FUNCTION()` --> `C.char`