libtch

package
v0.5.0 Latest Latest
Warning

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

Go to latest
Published: Nov 6, 2021 License: Apache-2.0 Imports: 7 Imported by: 2

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

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 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 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() int

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 AtcSetBenchmarkCudnn

func AtcSetBenchmarkCudnn(b int)

void atc_set_benchmark_cudnn(int b);

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 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 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 AtgAlias

func AtgAlias(ptr *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 AtgArangeOut

func AtgArangeOut(ptr *Ctensor, out Ctensor, end Cscalar)

func AtgArangeStart added in v0.4.0

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

func AtgArangeStartOut added in v0.4.0

func AtgArangeStartOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar)

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 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 AtgAsStrided

func AtgAsStrided(ptr *Ctensor, self 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)

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 AtgBartlettWindowPeriodic added in v0.4.0

func AtgBartlettWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, optionsDevice 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, meanDy Ctensor, meanDyXmu 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 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 AtgBatchNormGatherStatsWithCounts

func AtgBatchNormGatherStatsWithCounts(ptr *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 AtgBatchNormUpdateStats

func AtgBatchNormUpdateStats(ptr *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 AtgBernoulliOut

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

func AtgBernoulliP added in v0.4.0

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

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 AtgBinaryCrossEntropyWithLogitsBackward

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

func AtgBincount

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

func AtgBinomial added in v0.3.0

func AtgBinomial(ptr *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 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 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 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 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 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 AtgBlackmanWindowPeriodic added in v0.4.0

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

func AtgBlockDiag added in v0.3.0

func AtgBlockDiag(ptr *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 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_

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

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 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 AtgChannelShuffle added in v0.3.0

func AtgChannelShuffle(ptr *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 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 AtgCol2imBackward

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

func AtgCol2imBackwardGradInput added in v0.4.0

func AtgCol2imBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, 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 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 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 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 AtgConvDepthwise3dBackward added in v0.4.0

func AtgConvDepthwise3dBackward(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, 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 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 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 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 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 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 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 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 AtgCudnnConvolutionBackwardInput

func AtgCudnnConvolutionBackwardInput(ptr *Ctensor, selfSizeData []int64, selfSizeLen int, gradOutput Ctensor, weight Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32, allowTf32 int32)

func AtgCudnnConvolutionBackwardWeight

func AtgCudnnConvolutionBackwardWeight(ptr *Ctensor, weightSizeData []int64, weightSizeLen int, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32, allowTf32 int32)

func AtgCudnnConvolutionDeprecated added in v0.4.0

func AtgCudnnConvolutionDeprecated(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 AtgCudnnConvolutionDeprecated2 added in v0.4.0

func AtgCudnnConvolutionDeprecated2(ptr *Ctensor, self Ctensor, weight Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic 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 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 AtgCudnnConvolutionTransposeBackwardInput

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

func AtgCudnnConvolutionTransposeBackwardWeight

func AtgCudnnConvolutionTransposeBackwardWeight(ptr *Ctensor, weightSizeData []int64, weightSizeLen int, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32, allowTf32 int32)

func AtgCudnnConvolutionTransposeDeprecated added in v0.4.0

func AtgCudnnConvolutionTransposeDeprecated(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 AtgCudnnConvolutionTransposeDeprecated2 added in v0.4.0

func AtgCudnnConvolutionTransposeDeprecated2(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)

func AtgCudnnGridSampler

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

func AtgCudnnGridSamplerBackward

func AtgCudnnGridSamplerBackward(ptr *Ctensor, self Ctensor, grid Ctensor, gradOutput 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 AtgDetach_

func AtgDetach_(ptr *Ctensor, self Ctensor)

func AtgDiag

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

func AtgDiagBackward added in v0.3.0

func AtgDiagBackward(ptr *Ctensor, grad Ctensor, inputSizesData []int64, inputSizesLen int, diagonal int64)

func AtgDiagEmbed

func AtgDiagEmbed(ptr *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 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 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 AtgDivScalarMode_ added in v0.4.0

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

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 AtgEig

func AtgEig(ptr *Ctensor, self Ctensor, eigenvectors int32)

func AtgEigE added in v0.4.0

func AtgEigE(ptr *Ctensor, e Ctensor, v Ctensor, self Ctensor, eigenvectors int32)

func AtgEinsum

func AtgEinsum(ptr *Ctensor, equation string, tensorsData []Ctensor, tensorsLen 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 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 AtgEmptyOut

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

func AtgEmptyQuantized added in v0.3.0

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

func AtgEmptyStrided

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

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 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_

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 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 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 AtgFftHfftOut added in v0.4.0

func AtgFftHfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, 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 AtgFftIhfftOut added in v0.4.0

func AtgFftIhfftOut(ptr *Ctensor, out Ctensor, self Ctensor, nVal int64, nNull int, dim int64, 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 AtgFillDiagonal_

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

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 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)

func AtgFrobeniusNormDim added in v0.4.0

func AtgFrobeniusNormDim(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 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 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)

func AtgGeluBackward

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

func AtgGeluBackwardGradInput added in v0.5.0

func AtgGeluBackwardGradInput(ptr *Ctensor, gradInput Ctensor, grad Ctensor, self Ctensor)

func AtgGeluOut added in v0.5.0

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

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 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 AtgGridSampler2dBackward

func AtgGridSampler2dBackward(ptr *Ctensor, gradOutput 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 AtgGridSampler3dBackward

func AtgGridSampler3dBackward(ptr *Ctensor, gradOutput 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 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 AtgHannWindow

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

func AtgHannWindowPeriodic added in v0.4.0

func AtgHannWindowPeriodic(ptr *Ctensor, windowLength int64, periodic int32, optionsKind int32, optionsDevice 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 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 AtgIm2colBackward

func AtgIm2colBackward(ptr *Ctensor, gradOutput Ctensor, inputSizeData []int64, inputSizeLen int, kernelSizeData []int64, kernelSizeLen int, dilationData []int64, dilationLen int, paddingData []int64, paddingLen int, strideData []int64, strideLen int)

func AtgIm2colBackwardGradInput added in v0.4.0

func AtgIm2colBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, inputSizeData []int64, inputSizeLen int, 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 AtgIndexAddAlpha added in v0.4.0

func AtgIndexAddAlpha(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor, alpha Cscalar)

func AtgIndexAddAlpha_ added in v0.4.0

func AtgIndexAddAlpha_(ptr *Ctensor, self Ctensor, dim int64, index Ctensor, source Ctensor, alpha Cscalar)

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 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 AtgIndexFillIntTensor added in v0.4.0

func AtgIndexFillIntTensor(ptr *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 AtgIndexPut_

func AtgIndexPut_(ptr *Ctensor, self Ctensor, indicesData []Ctensor, indicesLen int, values Ctensor, accumulate 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 AtgIndices

func AtgIndices(ptr *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 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 AtgIsnan

func AtgIsnan(ptr *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 AtgKaiserWindowPeriodic added in v0.4.0

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

func AtgKlDiv

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

func AtgKlDivBackward

func AtgKlDivBackward(ptr *Ctensor, gradOutput 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 AtgL1LossBackward

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

func AtgL1LossBackwardGradInput added in v0.4.0

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

func AtgL1LossOut

func AtgL1LossOut(ptr *Ctensor, out 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 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 AtgLinalgDet added in v0.3.0

func AtgLinalgDet(ptr *Ctensor, self Ctensor)

func AtgLinalgDetOut added in v0.4.0

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

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, self Ctensor)

func AtgLinalgInvEx added in v0.4.0

func AtgLinalgInvEx(ptr *Ctensor, self Ctensor, checkErrors int32)

func AtgLinalgInvExInverse added in v0.4.0

func AtgLinalgInvExInverse(ptr *Ctensor, inverse Ctensor, info Ctensor, self Ctensor, checkErrors int32)

func AtgLinalgInvOut added in v0.4.0

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

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 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 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, tolVal float64, tolNull int, hermitian int32)

func AtgLinalgMatrixRankOut added in v0.4.0

func AtgLinalgMatrixRankOut(ptr *Ctensor, out Ctensor, self Ctensor, tolVal float64, tolNull int, 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 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, self Ctensor, mode string)

func AtgLinalgQrOut added in v0.4.0

func AtgLinalgQrOut(ptr *Ctensor, q Ctensor, r Ctensor, self Ctensor, mode string)

func AtgLinalgSlogdet added in v0.4.0

func AtgLinalgSlogdet(ptr *Ctensor, self Ctensor)

func AtgLinalgSlogdetOut added in v0.4.0

func AtgLinalgSlogdetOut(ptr *Ctensor, sign Ctensor, logabsdet Ctensor, self Ctensor)

func AtgLinalgSolve added in v0.4.0

func AtgLinalgSolve(ptr *Ctensor, input Ctensor, other Ctensor)

func AtgLinalgSolveOut added in v0.4.0

func AtgLinalgSolveOut(ptr *Ctensor, out Ctensor, input Ctensor, other Ctensor)

func AtgLinalgSvd added in v0.4.0

func AtgLinalgSvd(ptr *Ctensor, self Ctensor, fullMatrices int32)

func AtgLinalgSvdU added in v0.4.0

func AtgLinalgSvdU(ptr *Ctensor, u Ctensor, s Ctensor, vh Ctensor, self Ctensor, fullMatrices int32)

func AtgLinalgSvdvals added in v0.4.0

func AtgLinalgSvdvals(ptr *Ctensor, input Ctensor)

func AtgLinalgSvdvalsOut added in v0.4.0

func AtgLinalgSvdvalsOut(ptr *Ctensor, out Ctensor, input Ctensor)

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 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, stepsVal int64, stepsNull int, optionsKind int32, optionsDevice int32)

func AtgLinspaceOut

func AtgLinspaceOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar, stepsVal int64, stepsNull int)

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_

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 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, stepsVal int64, stepsNull int, base float64, optionsKind int32, optionsDevice int32)

func AtgLogspaceOut

func AtgLogspaceOut(ptr *Ctensor, out Ctensor, start Cscalar, end Cscalar, stepsVal int64, stepsNull int, 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 AtgLstsq

func AtgLstsq(ptr *Ctensor, self Ctensor, a Ctensor)

func AtgLstsqX added in v0.4.0

func AtgLstsqX(ptr *Ctensor, x Ctensor, qr Ctensor, self Ctensor, a Ctensor)

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 AtgMaskedFillTensor added in v0.4.0

func AtgMaskedFillTensor(ptr *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 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 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 AtgMatrixRank

func AtgMatrixRank(ptr *Ctensor, self Ctensor, symmetric int32)

func AtgMatrixRankTol added in v0.4.0

func AtgMatrixRankTol(ptr *Ctensor, self Ctensor, tol float64, symmetric int32)

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 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 AtgMaxUnpool2d

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

func AtgMaxUnpool2dBackward

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

func AtgMaxUnpool2dBackwardGradInput added in v0.4.0

func AtgMaxUnpool2dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput 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 AtgMaxUnpool3dBackward

func AtgMaxUnpool3dBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, indices Ctensor, outputSizeData []int64, outputSizeLen int, strideData []int64, strideLen int, paddingData []int64, paddingLen int)

func AtgMaxUnpool3dBackwardGradInput added in v0.4.0

func AtgMaxUnpool3dBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput 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 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 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 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 AtgMiopenConvolutionBackwardBias

func AtgMiopenConvolutionBackwardBias(ptr *Ctensor, gradOutput Ctensor)

func AtgMiopenConvolutionBackwardInput

func AtgMiopenConvolutionBackwardInput(ptr *Ctensor, selfSizeData []int64, selfSizeLen int, gradOutput Ctensor, weight Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32)

func AtgMiopenConvolutionBackwardWeight

func AtgMiopenConvolutionBackwardWeight(ptr *Ctensor, weightSizeData []int64, weightSizeLen int, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32)

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 AtgMiopenConvolutionTransposeBackwardInput

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

func AtgMiopenConvolutionTransposeBackwardWeight

func AtgMiopenConvolutionTransposeBackwardWeight(ptr *Ctensor, weightSizeData []int64, weightSizeLen int, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen 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 AtgMiopenDepthwiseConvolutionBackwardInput

func AtgMiopenDepthwiseConvolutionBackwardInput(ptr *Ctensor, selfSizeData []int64, selfSizeLen int, gradOutput Ctensor, weight Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, benchmark int32, deterministic int32)

func AtgMiopenDepthwiseConvolutionBackwardWeight

func AtgMiopenDepthwiseConvolutionBackwardWeight(ptr *Ctensor, weightSizeData []int64, weightSizeLen int, gradOutput Ctensor, self 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 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 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 AtgMkldnnConvolutionBackwardInput

func AtgMkldnnConvolutionBackwardInput(ptr *Ctensor, selfSizeData []int64, selfSizeLen int, gradOutput Ctensor, weight Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, biasDefined int32)

func AtgMkldnnConvolutionBackwardWeights

func AtgMkldnnConvolutionBackwardWeights(ptr *Ctensor, weightSizeData []int64, weightSizeLen int, gradOutput Ctensor, self Ctensor, paddingData []int64, paddingLen int, strideData []int64, strideLen int, dilationData []int64, dilationLen int, groups int64, biasDefined int32)

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 AtgMkldnnLinearBackwardWeights added in v0.4.0

func AtgMkldnnLinearBackwardWeights(ptr *Ctensor, gradOutput Ctensor, input Ctensor, weight Ctensor, biasDefined int32)

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 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 AtgMkldnnReorderConv2dWeight

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

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 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 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 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 AtgNanquantile added in v0.3.0

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

func AtgNanquantileNew added in v0.4.0

func AtgNanquantileNew(ptr *Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgNanquantileNewOut added in v0.4.0

func AtgNanquantileNewOut(ptr *Ctensor, out Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgNanquantileNewScalar added in v0.4.0

func AtgNanquantileNewScalar(ptr *Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgNanquantileNewScalarOut added in v0.4.0

func AtgNanquantileNewScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, q float64, 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)

func AtgNanquantileScalar added in v0.4.0

func AtgNanquantileScalar(ptr *Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32)

func AtgNanquantileScalarOut added in v0.4.0

func AtgNanquantileScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32)

func AtgNansum added in v0.3.0

func AtgNansum(ptr *Ctensor, self Ctensor, dtype int32)

func AtgNansumDimIntlist added in v0.4.0

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

func AtgNansumIntlistOut added in v0.4.0

func AtgNansumIntlistOut(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 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 AtgNativeLayerNorm

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

func AtgNativeNorm

func AtgNativeNorm(ptr *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 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 AtgNewEmpty

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

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 AtgNewFull

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

func AtgNewOnes added in v0.5.0

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

func AtgNewZeros

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

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 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 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 AtgNormal added in v0.4.0

func AtgNormal(ptr *Ctensor, out Ctensor, mean Ctensor, std float64)

func AtgNormalFloatFloatOut added in v0.4.0

func AtgNormalFloatFloatOut(ptr *Ctensor, out Ctensor, mean float64, std float64, sizeData []int64, sizeLen int)

func AtgNormalFloatTensorOut added in v0.4.0

func AtgNormalFloatTensorOut(ptr *Ctensor, out Ctensor, mean float64, std Ctensor)

func AtgNormalTensorTensorOut added in v0.4.0

func AtgNormalTensorTensorOut(ptr *Ctensor, out Ctensor, mean Ctensor, std Ctensor)

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 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 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 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 AtgPixelUnshuffle added in v0.4.0

func AtgPixelUnshuffle(ptr *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 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 AtgPreluBackward

func AtgPreluBackward(ptr *Ctensor, gradOutput 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 AtgPut added in v0.4.0

func AtgPut(ptr *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 AtgQPerChannelZeroPoints

func AtgQPerChannelZeroPoints(ptr *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)

func AtgQuantileNew added in v0.4.0

func AtgQuantileNew(ptr *Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgQuantileNewOut added in v0.4.0

func AtgQuantileNewOut(ptr *Ctensor, out Ctensor, self Ctensor, q Ctensor, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgQuantileNewScalar added in v0.4.0

func AtgQuantileNewScalar(ptr *Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32, interpolation string)

func AtgQuantileNewScalarOut added in v0.4.0

func AtgQuantileNewScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, q float64, 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)

func AtgQuantileScalar added in v0.4.0

func AtgQuantileScalar(ptr *Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32)

func AtgQuantileScalarOut added in v0.4.0

func AtgQuantileScalarOut(ptr *Ctensor, out Ctensor, self Ctensor, q float64, dimVal int64, dimNull int, keepdim int32)

func AtgQuantizePerChannel

func AtgQuantizePerChannel(ptr *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 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 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 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 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 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 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 AtgRandnOut

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

func AtgRandomFrom_ added in v0.4.0

func AtgRandomFrom_(ptr *Ctensor, self Ctensor, from int64, toVal int64, toNull int)

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 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 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 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 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 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 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 AtgRot90

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

func AtgRound

func AtgRound(ptr *Ctensor, self Ctensor)

func AtgRoundOut

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

func AtgRound_

func AtgRound_(ptr *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 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 AtgScalarTensor

func AtgScalarTensor(ptr *Ctensor, s Cscalar, optionsKind int32, optionsDevice int32)

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)

func AtgSearchsortedScalar added in v0.4.0

func AtgSearchsortedScalar(ptr *Ctensor, sortedSequence Ctensor, selfScalar Cscalar, outInt32 int32, right int32)

func AtgSearchsortedTensorOut added in v0.4.0

func AtgSearchsortedTensorOut(ptr *Ctensor, out Ctensor, sortedSequence Ctensor, self Ctensor, outInt32 int32, right int32)

func AtgSegmentReduce added in v0.4.0

func AtgSegmentReduce(ptr *Ctensor, data Ctensor, reduce string, lengths Ctensor, indices 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 AtgSelu

func AtgSelu(ptr *Ctensor, self Ctensor)

func AtgSelu_

func AtgSelu_(ptr *Ctensor, self Ctensor)

func AtgSetRequiresGrad

func AtgSetRequiresGrad(ptr *Ctensor, self Ctensor, r int32)

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 AtgSlogdet

func AtgSlogdet(ptr *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 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 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 AtgSoftplus

func AtgSoftplus(ptr *Ctensor, self Ctensor)

func AtgSoftplusBackward

func AtgSoftplusBackward(ptr *Ctensor, gradOutput Ctensor, self Ctensor, beta Cscalar, threshold Cscalar, output Ctensor)

func AtgSoftplusBackwardGradInput added in v0.4.0

func AtgSoftplusBackwardGradInput(ptr *Ctensor, gradInput Ctensor, gradOutput Ctensor, self Ctensor, beta Cscalar, threshold Cscalar, output Ctensor)

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 AtgSolve

func AtgSolve(ptr *Ctensor, self Ctensor, a Ctensor)

func AtgSolveSolution added in v0.4.0

func AtgSolveSolution(ptr *Ctensor, solution Ctensor, lu Ctensor, self Ctensor, a 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 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)

func AtgSparseCooTensorIndicesSize added in v0.4.0

func AtgSparseCooTensorIndicesSize(ptr *Ctensor, indices 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 AtgSparseResizeAndClear_

func AtgSparseResizeAndClear_(ptr *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 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 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 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 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 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)

func AtgSpecialRoundOut added in v0.5.0

func AtgSpecialRoundOut(ptr *Ctensor, out Ctensor, self 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 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 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 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, correctionVal int64, correctionNull int, keepdim int32)

func AtgStdCorrectionOut added in v0.4.0

func AtgStdCorrectionOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, correctionVal int64, correctionNull int, 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, correctionVal int64, correctionNull int, 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 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 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 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 AtgSymeig

func AtgSymeig(ptr *Ctensor, self Ctensor, eigenvectors int32, upper int32)

func AtgSymeigE added in v0.4.0

func AtgSymeigE(ptr *Ctensor, e Ctensor, v Ctensor, self Ctensor, eigenvectors int32, upper int32)

func AtgT

func AtgT(ptr *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)

func AtgToDenseBackward

func AtgToDenseBackward(ptr *Ctensor, grad Ctensor, input Ctensor)

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 AtgToOther added in v0.4.0

func AtgToOther(ptr *Ctensor, self Ctensor, other Ctensor, nonBlocking int32, copy int32)

func AtgToSparse

func AtgToSparse(ptr *Ctensor, self Ctensor)

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 AtgTranspose

func AtgTranspose(ptr *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 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 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 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 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 AtgUnsqueeze

func AtgUnsqueeze(ptr *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 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 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 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 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 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 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 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 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, correctionVal int64, correctionNull int, keepdim int32)

func AtgVarCorrectionOut added in v0.4.0

func AtgVarCorrectionOut(ptr *Ctensor, out Ctensor, self Ctensor, dimData []int64, dimLen int, correctionVal int64, correctionNull int, 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, correctionVal int64, correctionNull int, 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 AtgViewAsReal added in v0.3.0

func AtgViewAsReal(ptr *Ctensor, self Ctensor)

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 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_

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 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_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_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_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_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_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_BaddbmmMkl_

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

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_Cat

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

func Atg_CatOut

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

func Atg_CdistBackward

func Atg_CdistBackward(ptr *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_Coalesce added in v0.4.0

func Atg_Coalesce(ptr *Ctensor, self Ctensor)

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_ConjPhysical added in v0.5.0

func Atg_ConjPhysical(ptr *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_ConvDepthwise2dBackward added in v0.5.0

func Atg_ConvDepthwise2dBackward(ptr *Ctensor, gradInput Ctensor, gradWeight Ctensor, gradOutput Ctensor, self Ctensor, weight Ctensor, kernelSizeData []int64, kernelSizeLen int, 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_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_ConvolutionNogroup

func Atg_ConvolutionNogroup(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)

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_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_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_CudnnInitDropoutState

func Atg_CudnnInitDropoutState(ptr *Ctensor, dropout float64, train int32, dropoutSeed int64, optionsKind int32, optionsDevice int32)

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_CufftGetPlanCacheMaxSize added in v0.4.0

func Atg_CufftGetPlanCacheMaxSize(deviceIndex int64) int64

func Atg_CufftGetPlanCacheSize added in v0.4.0

func Atg_CufftGetPlanCacheSize(deviceIndex int64) int64

func Atg_DebugHasInternalOverlap added in v0.4.0

func Atg_DebugHasInternalOverlap(self Ctensor) int64

func Atg_DetLuBasedHelper added in v0.5.0

func Atg_DetLuBasedHelper(ptr *Ctensor, self Ctensor)

func Atg_DetLuBasedHelperBackwardHelper added in v0.5.0

func Atg_DetLuBasedHelperBackwardHelper(ptr *Ctensor, detGrad Ctensor, det Ctensor, self Ctensor, lu Ctensor, pivs Ctensor)

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_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_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_EmbeddingBagPerSampleWeightsBackward

func Atg_EmbeddingBagPerSampleWeightsBackward(ptr *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_EmptyPerChannelAffineQuantized

func Atg_EmptyPerChannelAffineQuantized(ptr *Ctensor, sizeData []int64, sizeLen int, scales Ctensor, zeroPoints Ctensor, axis int64, optionsKind int32, optionsDevice int32)

func Atg_EuclideanDist added in v0.3.0

func Atg_EuclideanDist(ptr *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_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_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_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_FusedDropout

func Atg_FusedDropout(ptr *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_FwPrimal added in v0.4.0

func Atg_FwPrimal(ptr *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_HasCompatibleShallowCopyType added in v0.4.0

func Atg_HasCompatibleShallowCopyType(self Ctensor, from Ctensor) bool

func Atg_IndexCopy_

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

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_InverseHelper

func Atg_InverseHelper(ptr *Ctensor, self Ctensor)

func Atg_LinalgInvOutHelper_ added in v0.4.0

func Atg_LinalgInvOutHelper_(ptr *Ctensor, self Ctensor, infosLu Ctensor, infosGetri Ctensor)

func Atg_LinalgQrHelper added in v0.4.0

func Atg_LinalgQrHelper(ptr *Ctensor, self Ctensor, mode 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, self Ctensor)

func Atg_LogSoftmaxBackwardDataOut added in v0.5.0

func Atg_LogSoftmaxBackwardDataOut(ptr *Ctensor, out Ctensor, gradOutput Ctensor, output Ctensor, dim int64, self Ctensor)

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_LuWithInfo

func Atg_LuWithInfo(ptr *Ctensor, self Ctensor, pivot int32, checkErrors int32)

func Atg_MakeDual added in v0.4.0

func Atg_MakeDual(ptr *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_MakePerTensorQuantizedTensor

func Atg_MakePerTensorQuantizedTensor(ptr *Ctensor, self Ctensor, scale float64, zeroPoint int64)

func Atg_MaskedScale

func Atg_MaskedScale(ptr *Ctensor, self Ctensor, mask Ctensor, scale float64)

func Atg_MkldnnReshape

func Atg_MkldnnReshape(ptr *Ctensor, self Ctensor, shapeData []int64, shapeLen int)

func Atg_MkldnnTranspose

func Atg_MkldnnTranspose(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)

func Atg_MkldnnTranspose_

func Atg_MkldnnTranspose_(ptr *Ctensor, self Ctensor, dim0 int64, dim1 int64)

func Atg_NegView added in v0.5.0

func Atg_NegView(ptr *Ctensor, self Ctensor)

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_NnpackSpatialConvolutionBackwardInput

func Atg_NnpackSpatialConvolutionBackwardInput(ptr *Ctensor, input Ctensor, gradOutput Ctensor, weight Ctensor, paddingData []int64, paddingLen int)

func Atg_NnpackSpatialConvolutionBackwardWeight

func Atg_NnpackSpatialConvolutionBackwardWeight(ptr *Ctensor, input Ctensor, weightsizeData []int64, weightsizeLen int, gradOutput Ctensor, paddingData []int64, paddingLen 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_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_PinMemory added in v0.5.0

func Atg_PinMemory(ptr *Ctensor, self Ctensor, device int32)

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_ReshapeFromTensor

func Atg_ReshapeFromTensor(ptr *Ctensor, self Ctensor, shape Ctensor)

func Atg_RowwisePrune added in v0.4.0

func Atg_RowwisePrune(ptr *Ctensor, weight Ctensor, mask Ctensor, compressedIndicesDtype int32)

func Atg_SWhere

func Atg_SWhere(ptr *Ctensor, condition Ctensor, self Ctensor, other Ctensor)

func Atg_SampleDirichlet

func Atg_SampleDirichlet(ptr *Ctensor, self Ctensor)

func Atg_SaturateWeightToFp16 added in v0.3.0

func Atg_SaturateWeightToFp16(ptr *Ctensor, weight Ctensor)

func Atg_SegmentReduceBackward added in v0.5.0

func Atg_SegmentReduceBackward(ptr *Ctensor, grad Ctensor, output Ctensor, data Ctensor, reduce string, lengths Ctensor, axis int64)

func Atg_ShapeAsTensor

func Atg_ShapeAsTensor(ptr *Ctensor, self Ctensor)

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, self Ctensor)

func Atg_SoftmaxBackwardDataOut added in v0.5.0

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

func Atg_SoftmaxOut added in v0.5.0

func Atg_SoftmaxOut(ptr *Ctensor, out Ctensor, self Ctensor, dim int64, halfToFloat int32)

func Atg_SolveHelper

func Atg_SolveHelper(ptr *Ctensor, self Ctensor, a Ctensor)

func Atg_SparseAddmm

func Atg_SparseAddmm(ptr *Ctensor, self Ctensor, sparse Ctensor, dense Ctensor)

func Atg_SparseCooTensorUnsafe

func Atg_SparseCooTensorUnsafe(ptr *Ctensor, indices Ctensor, values Ctensor, sizeData []int64, sizeLen int, optionsKind int32, optionsDevice 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)

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_SparseLogSoftmaxInt added in v0.4.0

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

func Atg_SparseMaskHelper added in v0.4.0

func Atg_SparseMaskHelper(ptr *Ctensor, t Ctensor, maskIndices Ctensor)

func Atg_SparseMm

func Atg_SparseMm(ptr *Ctensor, sparse Ctensor, dense Ctensor)

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_SparseSoftmaxInt added in v0.4.0

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

func Atg_SparseSparseMatmul added in v0.4.0

func Atg_SparseSparseMatmul(ptr *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_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_SparseSumDtype added in v0.4.0

func Atg_SparseSumDtype(ptr *Ctensor, self Ctensor, dtype int32)

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_SvdHelper

func Atg_SvdHelper(ptr *Ctensor, self Ctensor, some int32, computeUv int32)

func Atg_SymeigHelper

func Atg_SymeigHelper(ptr *Ctensor, self Ctensor, eigenvectors int32, upper int32)

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_TestOptionalFilledIntlist added in v0.3.0

func Atg_TestOptionalFilledIntlist(ptr *Ctensor, values Ctensor, addendsData []int64, addendsLen int)

func Atg_TestOptionalIntlist added in v0.3.0

func Atg_TestOptionalIntlist(ptr *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_ToCopy added in v0.5.0

func Atg_ToCopy(ptr *Ctensor, self Ctensor, optionsKind int32, optionsDevice int32, nonBlocking int32)

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_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_UnpackDual added in v0.4.0

func Atg_UnpackDual(ptr *Ctensor, dual Ctensor, level int64)

func Atg_UnsafeView

func Atg_UnsafeView(ptr *Ctensor, self Ctensor, sizeData []int64, sizeLen 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_UseCudnnRnnFlattenWeight added in v0.4.0

func Atg_UseCudnnRnnFlattenWeight() bool

func Atg_Values

func Atg_Values(ptr *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_WeightNormCudaInterface

func Atg_WeightNormCudaInterface(ptr *Ctensor, v Ctensor, g Ctensor, dim int64)

func Atg_WeightNormCudaInterfaceBackward

func Atg_WeightNormCudaInterfaceBackward(ptr *Ctensor, gradW Ctensor, savedV Ctensor, savedG Ctensor, savedNorms Ctensor, dim int64)

func Atg_WeightNormDifferentiableBackward

func Atg_WeightNormDifferentiableBackward(ptr *Ctensor, gradW Ctensor, savedV Ctensor, savedG Ctensor, savedNorms 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__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__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 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

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

Jump to

Keyboard shortcuts

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