Created
June 9, 2018 17:19
-
-
Save Mizzlr/7c5074e76d1e8af501ee1de5e069d5d4 to your computer and use it in GitHub Desktop.
torch/csrc/autograd/generated/VariableType.h for use in https://github.com/pytorch/pytorch/blob/master/torch/onnx/symbolic.py
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#pragma once | |
// @generated from tools/autograd/templates/VariableType.h | |
#include <ATen/ATen.h> | |
#include <cstdint> // for size_t | |
#include <functional> // for function | |
#include <memory> // for unique_ptr | |
#include <string> | |
#include <vector> | |
namespace torch { namespace autograd { | |
struct Variable; | |
using at::Context; | |
using at::Generator; | |
using at::IntList; | |
using at::Scalar; | |
using at::SparseTensorRef; | |
using at::Storage; | |
using at::Tensor; | |
using at::TensorList; | |
using at::Type; | |
using at::ScalarType; | |
using at::optional; | |
void register_variable_type_for(Context*, at::Backend, at::ScalarType); | |
struct VariableType final : public at::Type { | |
VariableType(Context* context, at::Type* baseType); | |
virtual at::ScalarType scalarType() const override; | |
virtual at::Backend backend() const override; | |
virtual bool is_cuda() const override; | |
virtual bool is_sparse() const override; | |
virtual bool is_distributed() const override; | |
virtual std::unique_ptr<at::Storage> storage() const override; | |
virtual std::unique_ptr<at::Storage> storage(size_t size) const override; | |
virtual std::unique_ptr<at::Storage> storageFromBlob(void * data, int64_t size, const std::function<void(void*)> & deleter) const override; | |
virtual std::unique_ptr<Storage> storageWithAllocator(int64_t size, std::unique_ptr<at::Allocator> allocator) const override; | |
virtual std::unique_ptr<at::Generator> generator() const override; | |
virtual const char * toString() const override; | |
virtual at::TypeID ID() const override; | |
virtual size_t elementSizeInBytes() const override; | |
virtual at::Type & toBackend(at::Backend b) const override; | |
virtual at::Type & toScalarType(at::ScalarType s) const override; | |
static const char * typeString(); | |
virtual std::unique_ptr<at::Storage> unsafeStorageFromTH(void * th_pointer, bool retain) const override; | |
virtual at::Tensor unsafeTensorFromTH(void * th_pointer, bool retain) const override; | |
static at::Type* getType(const at::Type& baseType); | |
static at::Type* getType(const at::Tensor& tensor); | |
static bool isVariableType(const at::Type& type); | |
static std::vector<at::Type*> allCUDATypes(); | |
static std::vector<at::Type*> allCPUTypes(); | |
virtual Tensor & s_copy_(Tensor & self, const Tensor & src, bool non_blocking) const override; | |
virtual Tensor & _s_copy_from(const Tensor & self, Tensor & dst, bool non_blocking) const override; | |
virtual int64_t storage_offset(const Tensor & self) const override; | |
virtual Tensor & resize_(Tensor & self, IntList size) const override; | |
virtual int64_t numel(const Tensor & self) const override; | |
virtual Tensor & set_(Tensor & self, Storage & storage) const override; | |
virtual Tensor & set_(Tensor & self, Storage & sourceStorage, int64_t storage_offset, IntList size, IntList stride) const override; | |
virtual Tensor & set_(Tensor & self, const Tensor & source) const override; | |
virtual Tensor & set_(Tensor & self) const override; | |
virtual Tensor & _fill_(Tensor & self, Scalar value) const override; | |
virtual Tensor & _fill_(Tensor & self, const Tensor & value) const override; | |
virtual bool is_contiguous(const Tensor & self) const override; | |
virtual bool is_set_to(const Tensor & self, const Tensor & tensor) const override; | |
virtual Tensor & s_masked_fill_(Tensor & self, const Tensor & mask, Scalar value) const override; | |
virtual Tensor & s_masked_fill_(Tensor & self, const Tensor & mask, const Tensor & value) const override; | |
virtual Tensor & s_masked_scatter_(Tensor & self, const Tensor & mask, const Tensor & source) const override; | |
virtual Tensor & s_masked_select_out(Tensor & result, const Tensor & self, const Tensor & mask) const override; | |
virtual Tensor s_masked_select(const Tensor & self, const Tensor & mask) const override; | |
virtual Tensor & nonzero_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor nonzero(const Tensor & self) const override; | |
virtual Tensor contiguous(const Tensor & self) const override; | |
virtual Tensor clone(const Tensor & self) const override; | |
virtual Tensor view(const Tensor & self, IntList size) const override; | |
virtual Tensor & resize_as_(Tensor & self, const Tensor & the_template) const override; | |
virtual Tensor & index_select_out(Tensor & result, const Tensor & self, int64_t dim, const Tensor & index) const override; | |
virtual Tensor index_select(const Tensor & self, int64_t dim, const Tensor & index) const override; | |
virtual Tensor & _indexCopy_(Tensor & self, int64_t dim, const Tensor & index, const Tensor & source) const override; | |
virtual Tensor & take_out(Tensor & result, const Tensor & self, const Tensor & index) const override; | |
virtual Tensor take(const Tensor & self, const Tensor & index) const override; | |
virtual Tensor & put_(Tensor & self, const Tensor & index, const Tensor & source, bool accumulate) const override; | |
virtual Tensor & index_add_(Tensor & self, int64_t dim, const Tensor & index, const Tensor & source) const override; | |
virtual Tensor & index_fill_(Tensor & self, int64_t dim, const Tensor & index, Scalar value) const override; | |
virtual Tensor & index_fill_(Tensor & self, int64_t dim, const Tensor & index, const Tensor & value) const override; | |
virtual Tensor unfold(const Tensor & self, int64_t dimension, int64_t size, int64_t step) const override; | |
virtual Tensor & _range_out(Tensor & result, Scalar start, Scalar end, Scalar step) const override; | |
virtual Tensor _range(Scalar start, Scalar end, Scalar step) const override; | |
virtual Tensor & _arange_out(Tensor & result, Scalar start, Scalar end, Scalar step) const override; | |
virtual Tensor _arange(Scalar start, Scalar end, Scalar step) const override; | |
virtual Tensor & _arange_out(Tensor & result, Scalar end) const override; | |
virtual Tensor _arange(Scalar end) const override; | |
virtual Tensor & scatter_(Tensor & self, int64_t dim, const Tensor & index, const Tensor & src) const override; | |
virtual Tensor & scatter_(Tensor & self, int64_t dim, const Tensor & index, Scalar value) const override; | |
virtual Tensor & scatter_add_(Tensor & self, int64_t dim, const Tensor & index, const Tensor & src) const override; | |
virtual Tensor & gather_out(Tensor & result, const Tensor & self, int64_t dim, const Tensor & index) const override; | |
virtual Tensor gather(const Tensor & self, int64_t dim, const Tensor & index) const override; | |
virtual void* data_ptr(const Tensor & self) const override; | |
virtual bool equal(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __and___out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor __and__(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s___and___out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s___and__(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __iand__(Tensor & self, Scalar other) const override; | |
virtual Tensor & s___iand__(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __or___out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor __or__(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s___or___out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s___or__(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __ior__(Tensor & self, Scalar other) const override; | |
virtual Tensor & s___ior__(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __xor___out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor __xor__(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s___xor___out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s___xor__(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __ixor__(Tensor & self, Scalar other) const override; | |
virtual Tensor & s___ixor__(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __lshift___out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor __lshift__(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s___lshift___out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s___lshift__(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __ilshift__(Tensor & self, Scalar other) const override; | |
virtual Tensor & s___ilshift__(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __rshift___out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor __rshift__(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s___rshift___out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s___rshift__(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & __irshift__(Tensor & self, Scalar other) const override; | |
virtual Tensor & s___irshift__(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & lt_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor lt(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_lt_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_lt(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & lt_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_lt_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & gt_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor gt(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_gt_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_gt(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & gt_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_gt_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & le_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor le(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_le_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_le(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & le_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_le_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & ge_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor ge(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_ge_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_ge(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & ge_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_ge_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & eq_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor eq(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_eq_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_eq(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & eq_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_eq_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & ne_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor ne(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_ne_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_ne(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & ne_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_ne_(Tensor & self, const Tensor & other) const override; | |
virtual std::tuple<Tensor &,Tensor &> min_out(Tensor & min, Tensor & min_indices, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual std::tuple<Tensor,Tensor> min(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor & s_min_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_min(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor min(const Tensor & self) const override; | |
virtual std::tuple<Tensor &,Tensor &> max_out(Tensor & max, Tensor & max_indices, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual std::tuple<Tensor,Tensor> max(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor & s_max_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_max(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor max(const Tensor & self) const override; | |
virtual std::tuple<Tensor &,Tensor &> kthvalue_out(Tensor & values, Tensor & indices, const Tensor & self, int64_t k, int64_t dim, bool keepdim) const override; | |
virtual std::tuple<Tensor,Tensor> kthvalue(const Tensor & self, int64_t k, int64_t dim, bool keepdim) const override; | |
virtual std::tuple<Tensor &,Tensor &> mode_out(Tensor & values, Tensor & indices, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual std::tuple<Tensor,Tensor> mode(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual std::tuple<Tensor &,Tensor &> median_out(Tensor & values, Tensor & indices, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual std::tuple<Tensor,Tensor> median(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor median(const Tensor & self) const override; | |
virtual std::tuple<Tensor &,Tensor &> sort_out(Tensor & values, Tensor & indices, const Tensor & self, int64_t dim, bool descending) const override; | |
virtual std::tuple<Tensor,Tensor> sort(const Tensor & self, int64_t dim, bool descending) const override; | |
virtual std::tuple<Tensor &,Tensor &> topk_out(Tensor & values, Tensor & indices, const Tensor & self, int64_t k, int64_t dim, bool largest, bool sorted) const override; | |
virtual std::tuple<Tensor,Tensor> topk(const Tensor & self, int64_t k, int64_t dim, bool largest, bool sorted) const override; | |
virtual Tensor & all_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor all(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor all(const Tensor & self) const override; | |
virtual Tensor & any_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor any(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor any(const Tensor & self) const override; | |
virtual int64_t get_device(const Tensor & self) const override; | |
virtual Tensor & _abs_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _abs(const Tensor & self) const override; | |
virtual Tensor & sigmoid_(Tensor & self) const override; | |
virtual Tensor & sigmoid_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor sigmoid(const Tensor & self) const override; | |
virtual Tensor & _log_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _log(const Tensor & self) const override; | |
virtual Tensor & _log10_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _log10(const Tensor & self) const override; | |
virtual Tensor & _log1p_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _log1p(const Tensor & self) const override; | |
virtual Tensor & _log2_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _log2(const Tensor & self) const override; | |
virtual Tensor & lgamma_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor lgamma(const Tensor & self) const override; | |
virtual Tensor & lgamma_(Tensor & self) const override; | |
virtual Tensor & digamma_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor digamma(const Tensor & self) const override; | |
virtual Tensor & digamma_(Tensor & self) const override; | |
virtual Tensor & polygamma_out(Tensor & result, int64_t n, const Tensor & self) const override; | |
virtual Tensor polygamma(int64_t n, const Tensor & self) const override; | |
virtual Tensor & polygamma_(Tensor & self, int64_t n) const override; | |
virtual Tensor & _exp_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _exp(const Tensor & self) const override; | |
virtual Tensor & _expm1_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _expm1(const Tensor & self) const override; | |
virtual Tensor & _cos_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _cos(const Tensor & self) const override; | |
virtual Tensor & _acos_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _acos(const Tensor & self) const override; | |
virtual Tensor & _cosh_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _cosh(const Tensor & self) const override; | |
virtual Tensor & _sin_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _sin(const Tensor & self) const override; | |
virtual Tensor & _asin_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _asin(const Tensor & self) const override; | |
virtual Tensor & _sinh_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _sinh(const Tensor & self) const override; | |
virtual Tensor & _tan_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _tan(const Tensor & self) const override; | |
virtual Tensor & _atan_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _atan(const Tensor & self) const override; | |
virtual Tensor & _th_tanh_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _th_tanh(const Tensor & self) const override; | |
virtual Tensor & _erf_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _erf(const Tensor & self) const override; | |
virtual Tensor & erfinv_(Tensor & self) const override; | |
virtual Tensor & erfinv_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor erfinv(const Tensor & self) const override; | |
virtual Tensor & _sqrt_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _sqrt(const Tensor & self) const override; | |
virtual Tensor & _rsqrt_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _rsqrt(const Tensor & self) const override; | |
virtual Tensor & _ceil_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _ceil(const Tensor & self) const override; | |
virtual Tensor & _floor_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _floor(const Tensor & self) const override; | |
virtual Tensor & _round_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _round(const Tensor & self) const override; | |
virtual Tensor & _trunc_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor _trunc(const Tensor & self) const override; | |
virtual Tensor & frac_(Tensor & self) const override; | |
virtual Tensor & frac_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor frac(const Tensor & self) const override; | |
virtual Tensor & mean_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor mean(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor mean(const Tensor & self) const override; | |
virtual Tensor & var_out(Tensor & result, const Tensor & self, int64_t dim, bool unbiased, bool keepdim) const override; | |
virtual Tensor var(const Tensor & self, int64_t dim, bool unbiased, bool keepdim) const override; | |
virtual Tensor var(const Tensor & self, bool unbiased) const override; | |
virtual Tensor & std_out(Tensor & result, const Tensor & self, int64_t dim, bool unbiased, bool keepdim) const override; | |
virtual Tensor std(const Tensor & self, int64_t dim, bool unbiased, bool keepdim) const override; | |
virtual Tensor std(const Tensor & self, bool unbiased) const override; | |
virtual Tensor & norm_out(Tensor & result, const Tensor & self, Scalar p, int64_t dim, bool keepdim) const override; | |
virtual Tensor norm(const Tensor & self, Scalar p, int64_t dim, bool keepdim) const override; | |
virtual Tensor norm(const Tensor & self, Scalar p) const override; | |
virtual Tensor & renorm_out(Tensor & result, const Tensor & self, Scalar p, int64_t dim, Scalar maxnorm) const override; | |
virtual Tensor renorm(const Tensor & self, Scalar p, int64_t dim, Scalar maxnorm) const override; | |
virtual Tensor & renorm_(Tensor & self, Scalar p, int64_t dim, Scalar maxnorm) const override; | |
virtual Tensor s_dist(const Tensor & self, const Tensor & other, Scalar p) const override; | |
virtual Tensor & reciprocal_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor reciprocal(const Tensor & self) const override; | |
virtual Tensor & reciprocal_(Tensor & self) const override; | |
virtual Tensor & neg_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor neg(const Tensor & self) const override; | |
virtual Tensor & neg_(Tensor & self) const override; | |
virtual Tensor & s_atan2_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_atan2(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & s_atan2_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & pow_out(Tensor & result, const Tensor & self, Scalar exponent) const override; | |
virtual Tensor pow(const Tensor & self, Scalar exponent) const override; | |
virtual Tensor & s_pow_out(Tensor & result, const Tensor & self, const Tensor & exponent) const override; | |
virtual Tensor s_pow(const Tensor & self, const Tensor & exponent) const override; | |
virtual Tensor & pow_out(Tensor & result, Scalar base, const Tensor & self) const override; | |
virtual Tensor pow(Scalar base, const Tensor & self) const override; | |
virtual Tensor & pow_(Tensor & self, Scalar exponent) const override; | |
virtual Tensor & s_pow_(Tensor & self, const Tensor & exponent) const override; | |
virtual Tensor & s_lerp_out(Tensor & result, const Tensor & self, const Tensor & end, Scalar weight) const override; | |
virtual Tensor s_lerp(const Tensor & self, const Tensor & end, Scalar weight) const override; | |
virtual Tensor & s_lerp_(Tensor & self, const Tensor & end, Scalar weight) const override; | |
virtual Tensor & _linspace_out(Tensor & result, Scalar start, Scalar end, int64_t steps) const override; | |
virtual Tensor _linspace(Scalar start, Scalar end, int64_t steps) const override; | |
virtual Tensor & _logspace_out(Tensor & result, Scalar start, Scalar end, int64_t steps) const override; | |
virtual Tensor _logspace(Scalar start, Scalar end, int64_t steps) const override; | |
virtual Tensor & histc_out(Tensor & result, const Tensor & self, int64_t bins, Scalar min, Scalar max) const override; | |
virtual Tensor histc(const Tensor & self, int64_t bins, Scalar min, Scalar max) const override; | |
virtual Tensor & zero_(Tensor & self) const override; | |
virtual Tensor _sumall(const Tensor & self) const override; | |
virtual Tensor & _th_sum_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor _th_sum(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor _prodall(const Tensor & self) const override; | |
virtual Tensor & _th_prod_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor _th_prod(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor & _cumsum_out(Tensor & result, const Tensor & self, int64_t dim) const override; | |
virtual Tensor _cumsum(const Tensor & self, int64_t dim) const override; | |
virtual Tensor & _cumprod_out(Tensor & result, const Tensor & self, int64_t dim) const override; | |
virtual Tensor _cumprod(const Tensor & self, int64_t dim) const override; | |
virtual Tensor & sign_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor sign(const Tensor & self) const override; | |
virtual Tensor & sign_(Tensor & self) const override; | |
virtual Tensor trace(const Tensor & self) const override; | |
virtual Tensor & add_out(Tensor & result, const Tensor & self, Scalar other, Scalar alpha) const override; | |
virtual Tensor add(const Tensor & self, Scalar other, Scalar alpha) const override; | |
virtual Tensor & s_add_out(Tensor & result, const Tensor & self, const Tensor & other, Scalar alpha) const override; | |
virtual Tensor s_add(const Tensor & self, const Tensor & other, Scalar alpha) const override; | |
virtual Tensor & add_out(Tensor & result, const Tensor & self, SparseTensorRef other, Scalar alpha) const override; | |
virtual Tensor add(const Tensor & self, SparseTensorRef other, Scalar alpha) const override; | |
virtual Tensor & add_(Tensor & self, Scalar other, Scalar alpha) const override; | |
virtual Tensor & s_add_(Tensor & self, const Tensor & other, Scalar alpha) const override; | |
virtual Tensor & add_(Tensor & self, SparseTensorRef other, Scalar alpha) const override; | |
virtual Tensor & sub_out(Tensor & result, const Tensor & self, Scalar other, Scalar alpha) const override; | |
virtual Tensor sub(const Tensor & self, Scalar other, Scalar alpha) const override; | |
virtual Tensor & s_sub_out(Tensor & result, const Tensor & self, const Tensor & other, Scalar alpha) const override; | |
virtual Tensor s_sub(const Tensor & self, const Tensor & other, Scalar alpha) const override; | |
virtual Tensor & sub_(Tensor & self, Scalar other, Scalar alpha) const override; | |
virtual Tensor & s_sub_(Tensor & self, const Tensor & other, Scalar alpha) const override; | |
virtual Tensor & mul_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor mul(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_mul_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_mul(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & mul_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_mul_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & div_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor div(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_div_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_div(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & div_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_div_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & fmod_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor fmod(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_fmod_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_fmod(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & fmod_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_fmod_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & remainder_out(Tensor & result, const Tensor & self, Scalar other) const override; | |
virtual Tensor remainder(const Tensor & self, Scalar other) const override; | |
virtual Tensor & s_remainder_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor s_remainder(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & remainder_(Tensor & self, Scalar other) const override; | |
virtual Tensor & s_remainder_(Tensor & self, const Tensor & other) const override; | |
virtual Tensor & clamp_out(Tensor & result, const Tensor & self, Scalar min, Scalar max) const override; | |
virtual Tensor clamp(const Tensor & self, Scalar min, Scalar max) const override; | |
virtual Tensor & clamp_(Tensor & self, Scalar min, Scalar max) const override; | |
virtual Tensor & clamp_min_out(Tensor & result, const Tensor & self, Scalar min) const override; | |
virtual Tensor clamp_min(const Tensor & self, Scalar min) const override; | |
virtual Tensor & clamp_min_(Tensor & self, Scalar min) const override; | |
virtual Tensor & clamp_max_out(Tensor & result, const Tensor & self, Scalar max) const override; | |
virtual Tensor clamp_max(const Tensor & self, Scalar max) const override; | |
virtual Tensor & clamp_max_(Tensor & self, Scalar max) const override; | |
virtual Tensor _dot(const Tensor & self, const Tensor & tensor) const override; | |
virtual Tensor & tril_out(Tensor & result, const Tensor & self, int64_t diagonal) const override; | |
virtual Tensor tril(const Tensor & self, int64_t diagonal) const override; | |
virtual Tensor & tril_(Tensor & self, int64_t diagonal) const override; | |
virtual Tensor & triu_out(Tensor & result, const Tensor & self, int64_t diagonal) const override; | |
virtual Tensor triu(const Tensor & self, int64_t diagonal) const override; | |
virtual Tensor & triu_(Tensor & self, int64_t diagonal) const override; | |
virtual Tensor & cross_out(Tensor & result, const Tensor & self, const Tensor & other, int64_t dim) const override; | |
virtual Tensor cross(const Tensor & self, const Tensor & other, int64_t dim) const override; | |
virtual Tensor & diag_out(Tensor & result, const Tensor & self, int64_t diagonal) const override; | |
virtual Tensor diag(const Tensor & self, int64_t diagonal) const override; | |
virtual Tensor & s_addmm_out(Tensor & result, const Tensor & self, const Tensor & mat1, const Tensor & mat2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor s_addmm(const Tensor & self, const Tensor & mat1, const Tensor & mat2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & addmm_out(Tensor & result, const Tensor & self, SparseTensorRef mat1, const Tensor & mat2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor addmm(const Tensor & self, SparseTensorRef mat1, const Tensor & mat2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & addmm_(Tensor & self, const Tensor & mat1, const Tensor & mat2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & addmm_(Tensor & self, SparseTensorRef mat1, const Tensor & mat2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & s__addmv_out(Tensor & result, const Tensor & self, const Tensor & mat, const Tensor & vec, Scalar beta, Scalar alpha) const override; | |
virtual Tensor s__addmv(const Tensor & self, const Tensor & mat, const Tensor & vec, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & _addmv_(Tensor & self, const Tensor & mat, const Tensor & vec, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & s__addr_out(Tensor & result, const Tensor & self, const Tensor & vec1, const Tensor & vec2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor s__addr(const Tensor & self, const Tensor & vec1, const Tensor & vec2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & _addr_(Tensor & self, const Tensor & vec1, const Tensor & vec2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & _ger_out(Tensor & result, const Tensor & self, const Tensor & vec2) const override; | |
virtual Tensor _ger(const Tensor & self, const Tensor & vec2) const override; | |
virtual Tensor & _mv_out(Tensor & result, const Tensor & self, const Tensor & vec) const override; | |
virtual Tensor _mv(const Tensor & self, const Tensor & vec) const override; | |
virtual Tensor & _mm_out(Tensor & result, const Tensor & self, const Tensor & mat2) const override; | |
virtual Tensor _mm(const Tensor & self, const Tensor & mat2) const override; | |
virtual Tensor & bmm_out(Tensor & result, const Tensor & self, const Tensor & mat2) const override; | |
virtual Tensor bmm(const Tensor & self, const Tensor & mat2) const override; | |
virtual Tensor & s_addbmm_out(Tensor & result, const Tensor & self, const Tensor & batch1, const Tensor & batch2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor s_addbmm(const Tensor & self, const Tensor & batch1, const Tensor & batch2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & addbmm_(Tensor & self, const Tensor & batch1, const Tensor & batch2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & s_baddbmm_out(Tensor & result, const Tensor & self, const Tensor & batch1, const Tensor & batch2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor s_baddbmm(const Tensor & self, const Tensor & batch1, const Tensor & batch2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & baddbmm_(Tensor & self, const Tensor & batch1, const Tensor & batch2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & s_addcmul_out(Tensor & result, const Tensor & self, const Tensor & tensor1, const Tensor & tensor2, Scalar value) const override; | |
virtual Tensor s_addcmul(const Tensor & self, const Tensor & tensor1, const Tensor & tensor2, Scalar value) const override; | |
virtual Tensor & s_addcmul_(Tensor & self, const Tensor & tensor1, const Tensor & tensor2, Scalar value) const override; | |
virtual Tensor & s_addcdiv_out(Tensor & result, const Tensor & self, const Tensor & tensor1, const Tensor & tensor2, Scalar value) const override; | |
virtual Tensor s_addcdiv(const Tensor & self, const Tensor & tensor1, const Tensor & tensor2, Scalar value) const override; | |
virtual Tensor & s_addcdiv_(Tensor & self, const Tensor & tensor1, const Tensor & tensor2, Scalar value) const override; | |
virtual std::tuple<Tensor &,Tensor &> _gesv_single_out(Tensor & solution, Tensor & lu, const Tensor & self, const Tensor & A) const override; | |
virtual std::tuple<Tensor,Tensor> _gesv_single(const Tensor & self, const Tensor & A) const override; | |
virtual std::tuple<Tensor &,Tensor &> gels_out(Tensor & res1, Tensor & res2, const Tensor & self, const Tensor & A) const override; | |
virtual std::tuple<Tensor,Tensor> gels(const Tensor & self, const Tensor & A) const override; | |
virtual std::tuple<Tensor &,Tensor &> trtrs_out(Tensor & res1, Tensor & res2, const Tensor & self, const Tensor & A, bool upper, bool transpose, bool unitriangular) const override; | |
virtual std::tuple<Tensor,Tensor> trtrs(const Tensor & self, const Tensor & A, bool upper, bool transpose, bool unitriangular) const override; | |
virtual std::tuple<Tensor &,Tensor &> symeig_out(Tensor & res1, Tensor & res2, const Tensor & self, bool eigenvectors, bool upper) const override; | |
virtual std::tuple<Tensor,Tensor> symeig(const Tensor & self, bool eigenvectors, bool upper) const override; | |
virtual std::tuple<Tensor &,Tensor &> eig_out(Tensor & res1, Tensor & res2, const Tensor & self, bool eigenvectors) const override; | |
virtual std::tuple<Tensor,Tensor> eig(const Tensor & self, bool eigenvectors) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> svd_out(Tensor & res1, Tensor & res2, Tensor & res3, const Tensor & self, bool some) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> svd(const Tensor & self, bool some) const override; | |
virtual Tensor & inverse_out(Tensor & output, const Tensor & self) const override; | |
virtual Tensor inverse(const Tensor & self) const override; | |
virtual Tensor & potrf_out(Tensor & output, const Tensor & self, bool upper) const override; | |
virtual Tensor potrf(const Tensor & self, bool upper) const override; | |
virtual Tensor & potrs_out(Tensor & result, const Tensor & self, const Tensor & input2, bool upper) const override; | |
virtual Tensor potrs(const Tensor & self, const Tensor & input2, bool upper) const override; | |
virtual Tensor & potri_out(Tensor & output, const Tensor & self, bool upper) const override; | |
virtual Tensor potri(const Tensor & self, bool upper) const override; | |
virtual std::tuple<Tensor &,Tensor &> pstrf_out(Tensor & res1, Tensor & res2, const Tensor & self, bool upper, Scalar tol) const override; | |
virtual std::tuple<Tensor,Tensor> pstrf(const Tensor & self, bool upper, Scalar tol) const override; | |
virtual std::tuple<Tensor &,Tensor &> qr_out(Tensor & res1, Tensor & res2, const Tensor & self) const override; | |
virtual std::tuple<Tensor,Tensor> qr(const Tensor & self) const override; | |
virtual std::tuple<Tensor &,Tensor &> geqrf_out(Tensor & res1, Tensor & res2, const Tensor & self) const override; | |
virtual std::tuple<Tensor,Tensor> geqrf(const Tensor & self) const override; | |
virtual Tensor & orgqr_out(Tensor & result, const Tensor & self, const Tensor & input2) const override; | |
virtual Tensor orgqr(const Tensor & self, const Tensor & input2) const override; | |
virtual Tensor & ormqr_out(Tensor & result, const Tensor & self, const Tensor & input2, const Tensor & input3, bool left, bool transpose) const override; | |
virtual Tensor ormqr(const Tensor & self, const Tensor & input2, const Tensor & input3, bool left, bool transpose) const override; | |
virtual std::tuple<Tensor &,Tensor &> btrifact_out(Tensor & result, Tensor & pivots, const Tensor & self, bool pivot) const override; | |
virtual std::tuple<Tensor,Tensor> btrifact(const Tensor & self, bool pivot) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> btrifact_with_info_out(Tensor & result, Tensor & pivots, Tensor & info, const Tensor & self, bool pivot) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> btrifact_with_info(const Tensor & self, bool pivot) const override; | |
virtual Tensor & btrisolve_out(Tensor & result, const Tensor & self, const Tensor & LU_data, const Tensor & LU_pivots) const override; | |
virtual Tensor btrisolve(const Tensor & self, const Tensor & LU_data, const Tensor & LU_pivots) const override; | |
virtual Tensor & random_(Tensor & self, int64_t from, int64_t to, Generator * generator) const override; | |
virtual Tensor & random_(Tensor & self, int64_t to, Generator * generator) const override; | |
virtual Tensor & random_(Tensor & self, Generator * generator) const override; | |
virtual Tensor & multinomial_out(Tensor & result, const Tensor & self, int64_t num_samples, bool replacement, Generator * generator) const override; | |
virtual Tensor multinomial(const Tensor & self, int64_t num_samples, bool replacement, Generator * generator) const override; | |
virtual Tensor & uniform_(Tensor & self, double from, double to, Generator * generator) const override; | |
virtual Tensor & normal_out(Tensor & output, const Tensor & mean, double std, Generator * generator) const override; | |
virtual Tensor normal(const Tensor & mean, double std, Generator * generator) const override; | |
virtual Tensor & normal_out(Tensor & output, double mean, const Tensor & std, Generator * generator) const override; | |
virtual Tensor normal(double mean, const Tensor & std, Generator * generator) const override; | |
virtual Tensor & normal_out(Tensor & output, const Tensor & mean, const Tensor & std, Generator * generator) const override; | |
virtual Tensor normal(const Tensor & mean, const Tensor & std, Generator * generator) const override; | |
virtual Tensor & normal_(Tensor & self, double mean, double std, Generator * generator) const override; | |
virtual Tensor & cauchy_(Tensor & self, double median, double sigma, Generator * generator) const override; | |
virtual Tensor & log_normal_(Tensor & self, double mean, double std, Generator * generator) const override; | |
virtual Tensor & exponential_(Tensor & self, double lambd, Generator * generator) const override; | |
virtual Tensor & geometric_(Tensor & self, double p, Generator * generator) const override; | |
virtual Tensor & _cpu_bernoulli_(Tensor & self, double p, Generator * generator) const override; | |
virtual Tensor & _th_bernoulli_out(Tensor & output, const Tensor & self, Generator * generator) const override; | |
virtual Tensor _th_bernoulli(const Tensor & self, Generator * generator) const override; | |
virtual Tensor & _dirichlet_grad_out(Tensor & output, const Tensor & x, const Tensor & alpha, const Tensor & total) const override; | |
virtual Tensor _dirichlet_grad(const Tensor & x, const Tensor & alpha, const Tensor & total) const override; | |
virtual Tensor tensor(Storage & storage, int64_t storageOffset, IntList size, IntList stride) const override; | |
virtual Tensor tensor(IntList size) const override; | |
virtual Tensor tensor(IntList size, IntList stride) const override; | |
virtual Tensor tensor() const override; | |
virtual Tensor sparse_coo_tensor(const Tensor & indices, const Tensor & values, IntList size) const override; | |
virtual Tensor sparse_coo_tensor(const Tensor & indices, const Tensor & values) const override; | |
virtual Tensor alias(const Tensor & self) const override; | |
virtual Tensor _sparse_coo_tensor_unsafe(const Tensor & indices, const Tensor & values, IntList size) const override; | |
virtual Tensor & _copy_ignoring_overlaps_(Tensor & self, const Tensor & src) const override; | |
virtual Tensor & as_strided_out(Tensor & result, const Tensor & self, IntList size, IntList stride, int64_t storage_offset) const override; | |
virtual Tensor as_strided(const Tensor & self, IntList size, IntList stride, int64_t storage_offset) const override; | |
virtual Tensor & as_strided_(Tensor & self, IntList size, IntList stride, int64_t storage_offset) const override; | |
virtual Tensor & sparse_raw_resize_(Tensor & self, IntList size, int64_t nDimI, int64_t nDimV) const override; | |
virtual Tensor & _cat_out(Tensor & self, TensorList tensors, int64_t dim) const override; | |
virtual Tensor _cat(TensorList tensors, int64_t dim) const override; | |
virtual Tensor _sparse_mask(const Tensor & self, SparseTensorRef mask) const override; | |
virtual Tensor to_dense(const Tensor & self) const override; | |
virtual int64_t _dimI(const Tensor & self) const override; | |
virtual int64_t _dimV(const Tensor & self) const override; | |
virtual int64_t _nnz(const Tensor & self) const override; | |
virtual Tensor coalesce(const Tensor & self) const override; | |
virtual bool is_coalesced(const Tensor & self) const override; | |
virtual Tensor _indices(const Tensor & self) const override; | |
virtual Tensor _values(const Tensor & self) const override; | |
virtual Tensor & hspmm_out(Tensor & result, const Tensor & mat1, const Tensor & mat2) const override; | |
virtual Tensor hspmm(const Tensor & mat1, const Tensor & mat2) const override; | |
virtual Tensor & binary_cross_entropy_out(Tensor & output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor binary_cross_entropy(const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor & binary_cross_entropy_forward_out(Tensor & output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor binary_cross_entropy_forward(const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor & binary_cross_entropy_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor binary_cross_entropy_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor & kl_div_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor kl_div(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & kl_div_forward_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor kl_div_forward(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & kl_div_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor kl_div_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & l1_loss_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor l1_loss(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & l1_loss_forward_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor l1_loss_forward(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & l1_loss_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor l1_loss_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & mse_loss_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor mse_loss(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & mse_loss_forward_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor mse_loss_forward(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & mse_loss_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor mse_loss_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & multi_margin_loss_out(Tensor & output, const Tensor & self, const Tensor & target, Scalar p, Scalar margin, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor multi_margin_loss(const Tensor & self, const Tensor & target, Scalar p, Scalar margin, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor & multi_margin_loss_forward_out(Tensor & output, const Tensor & self, const Tensor & target, Scalar p, Scalar margin, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor multi_margin_loss_forward(const Tensor & self, const Tensor & target, Scalar p, Scalar margin, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor & multi_margin_loss_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, Scalar p, Scalar margin, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor multi_margin_loss_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, Scalar p, Scalar margin, const Tensor & weight, bool size_average, bool reduce) const override; | |
virtual Tensor & multilabel_margin_loss_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor multilabel_margin_loss(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual std::tuple<Tensor &,Tensor &> multilabel_margin_loss_forward_out(Tensor & output, Tensor & is_target, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual std::tuple<Tensor,Tensor> multilabel_margin_loss_forward(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & multilabel_margin_loss_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce, const Tensor & is_target) const override; | |
virtual Tensor multilabel_margin_loss_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce, const Tensor & is_target) const override; | |
virtual Tensor & nll_loss_out(Tensor & output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce) const override; | |
virtual Tensor nll_loss(const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce) const override; | |
virtual std::tuple<Tensor &,Tensor &> nll_loss_forward_out(Tensor & output, Tensor & total_weight, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce) const override; | |
virtual std::tuple<Tensor,Tensor> nll_loss_forward(const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce) const override; | |
virtual Tensor & nll_loss_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce, const Tensor & total_weight) const override; | |
virtual Tensor nll_loss_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce, const Tensor & total_weight) const override; | |
virtual Tensor & nll_loss2d_out(Tensor & output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce) const override; | |
virtual Tensor nll_loss2d(const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce) const override; | |
virtual std::tuple<Tensor &,Tensor &> nll_loss2d_forward_out(Tensor & output, Tensor & total_weight, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce) const override; | |
virtual std::tuple<Tensor,Tensor> nll_loss2d_forward(const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce) const override; | |
virtual Tensor & nll_loss2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce, const Tensor & total_weight) const override; | |
virtual Tensor nll_loss2d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, const Tensor & weight, bool size_average, int64_t ignore_index, bool reduce, const Tensor & total_weight) const override; | |
virtual Tensor & smooth_l1_loss_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor smooth_l1_loss(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & smooth_l1_loss_forward_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor smooth_l1_loss_forward(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & smooth_l1_loss_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor smooth_l1_loss_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & soft_margin_loss_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor soft_margin_loss(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & soft_margin_loss_forward_out(Tensor & output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor soft_margin_loss_forward(const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & soft_margin_loss_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor soft_margin_loss_backward(const Tensor & grad_output, const Tensor & self, const Tensor & target, bool size_average, bool reduce) const override; | |
virtual Tensor & elu_out(Tensor & output, const Tensor & self, Scalar alpha, Scalar scale) const override; | |
virtual Tensor elu(const Tensor & self, Scalar alpha, Scalar scale) const override; | |
virtual Tensor & elu_forward_out(Tensor & output, const Tensor & self, Scalar alpha, Scalar scale) const override; | |
virtual Tensor elu_forward(const Tensor & self, Scalar alpha, Scalar scale) const override; | |
virtual Tensor & elu_backward_out(Tensor & grad_input, const Tensor & grad_output, Scalar alpha, Scalar scale, const Tensor & output) const override; | |
virtual Tensor elu_backward(const Tensor & grad_output, Scalar alpha, Scalar scale, const Tensor & output) const override; | |
virtual Tensor & elu_(Tensor & self, Scalar alpha, Scalar scale) const override; | |
virtual Tensor & elu_forward_(Tensor & self, Scalar alpha, Scalar scale) const override; | |
virtual Tensor & glu_out(Tensor & output, const Tensor & self, int64_t dim) const override; | |
virtual Tensor glu(const Tensor & self, int64_t dim) const override; | |
virtual Tensor & glu_forward_out(Tensor & output, const Tensor & self, int64_t dim) const override; | |
virtual Tensor glu_forward(const Tensor & self, int64_t dim) const override; | |
virtual Tensor & glu_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, int64_t dim) const override; | |
virtual Tensor glu_backward(const Tensor & grad_output, const Tensor & self, int64_t dim) const override; | |
virtual Tensor & hardshrink_out(Tensor & output, const Tensor & self, Scalar lambd) const override; | |
virtual Tensor hardshrink(const Tensor & self, Scalar lambd) const override; | |
virtual Tensor & hardshrink_forward_out(Tensor & output, const Tensor & self, Scalar lambd) const override; | |
virtual Tensor hardshrink_forward(const Tensor & self, Scalar lambd) const override; | |
virtual Tensor & hardshrink_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, Scalar lambd) const override; | |
virtual Tensor hardshrink_backward(const Tensor & grad_output, const Tensor & self, Scalar lambd) const override; | |
virtual Tensor & hardtanh_out(Tensor & output, const Tensor & self, Scalar min_val, Scalar max_val) const override; | |
virtual Tensor hardtanh(const Tensor & self, Scalar min_val, Scalar max_val) const override; | |
virtual Tensor & hardtanh_forward_out(Tensor & output, const Tensor & self, Scalar min_val, Scalar max_val) const override; | |
virtual Tensor hardtanh_forward(const Tensor & self, Scalar min_val, Scalar max_val) const override; | |
virtual Tensor & hardtanh_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, Scalar min_val, Scalar max_val) const override; | |
virtual Tensor hardtanh_backward(const Tensor & grad_output, const Tensor & self, Scalar min_val, Scalar max_val) const override; | |
virtual Tensor & hardtanh_(Tensor & self, Scalar min_val, Scalar max_val) const override; | |
virtual Tensor & hardtanh_forward_(Tensor & self, Scalar min_val, Scalar max_val) const override; | |
virtual Tensor & leaky_relu_out(Tensor & output, const Tensor & self, Scalar negative_slope) const override; | |
virtual Tensor leaky_relu(const Tensor & self, Scalar negative_slope) const override; | |
virtual Tensor & leaky_relu_forward_out(Tensor & output, const Tensor & self, Scalar negative_slope) const override; | |
virtual Tensor leaky_relu_forward(const Tensor & self, Scalar negative_slope) const override; | |
virtual Tensor & leaky_relu_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, Scalar negative_slope) const override; | |
virtual Tensor leaky_relu_backward(const Tensor & grad_output, const Tensor & self, Scalar negative_slope) const override; | |
virtual Tensor & leaky_relu_(Tensor & self, Scalar negative_slope) const override; | |
virtual Tensor & leaky_relu_forward_(Tensor & self, Scalar negative_slope) const override; | |
virtual Tensor & log_sigmoid_out(Tensor & output, const Tensor & self) const override; | |
virtual Tensor log_sigmoid(const Tensor & self) const override; | |
virtual std::tuple<Tensor &,Tensor &> log_sigmoid_forward_out(Tensor & output, Tensor & buffer, const Tensor & self) const override; | |
virtual std::tuple<Tensor,Tensor> log_sigmoid_forward(const Tensor & self) const override; | |
virtual Tensor & log_sigmoid_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & buffer) const override; | |
virtual Tensor log_sigmoid_backward(const Tensor & grad_output, const Tensor & self, const Tensor & buffer) const override; | |
virtual Tensor & prelu_out(Tensor & output, const Tensor & self, const Tensor & weight) const override; | |
virtual Tensor prelu(const Tensor & self, const Tensor & weight) const override; | |
virtual Tensor & prelu_forward_out(Tensor & output, const Tensor & self, const Tensor & weight) const override; | |
virtual Tensor prelu_forward(const Tensor & self, const Tensor & weight) const override; | |
virtual std::tuple<Tensor &,Tensor &> prelu_backward_out(Tensor & grad_input, Tensor & grad_weight, const Tensor & grad_output, const Tensor & self, const Tensor & weight) const override; | |
virtual std::tuple<Tensor,Tensor> prelu_backward(const Tensor & grad_output, const Tensor & self, const Tensor & weight, std::array<bool,2> output_mask) const override; | |
virtual Tensor & rrelu_with_noise_out(Tensor & output, const Tensor & self, const Tensor & noise, Scalar lower, Scalar upper, bool training, Generator * generator) const override; | |
virtual Tensor rrelu_with_noise(const Tensor & self, const Tensor & noise, Scalar lower, Scalar upper, bool training, Generator * generator) const override; | |
virtual Tensor & rrelu_with_noise_forward_out(Tensor & output, const Tensor & self, const Tensor & noise, Scalar lower, Scalar upper, bool training, Generator * generator) const override; | |
virtual Tensor rrelu_with_noise_forward(const Tensor & self, const Tensor & noise, Scalar lower, Scalar upper, bool training, Generator * generator) const override; | |
virtual Tensor & rrelu_with_noise_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & noise, Scalar lower, Scalar upper, bool training) const override; | |
virtual Tensor rrelu_with_noise_backward(const Tensor & grad_output, const Tensor & self, const Tensor & noise, Scalar lower, Scalar upper, bool training) const override; | |
virtual Tensor & rrelu_with_noise_(Tensor & self, const Tensor & noise, Scalar lower, Scalar upper, bool training, Generator * generator) const override; | |
virtual Tensor & rrelu_with_noise_forward_(Tensor & self, const Tensor & noise, Scalar lower, Scalar upper, bool training, Generator * generator) const override; | |
virtual Tensor & softplus_out(Tensor & output, const Tensor & self, Scalar beta, Scalar threshold) const override; | |
virtual Tensor softplus(const Tensor & self, Scalar beta, Scalar threshold) const override; | |
virtual Tensor & softplus_forward_out(Tensor & output, const Tensor & self, Scalar beta, Scalar threshold) const override; | |
virtual Tensor softplus_forward(const Tensor & self, Scalar beta, Scalar threshold) const override; | |
virtual Tensor & softplus_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, Scalar beta, Scalar threshold, const Tensor & output) const override; | |
virtual Tensor softplus_backward(const Tensor & grad_output, const Tensor & self, Scalar beta, Scalar threshold, const Tensor & output) const override; | |
virtual Tensor & softshrink_out(Tensor & output, const Tensor & self, Scalar lambd) const override; | |
virtual Tensor softshrink(const Tensor & self, Scalar lambd) const override; | |
virtual Tensor & softshrink_forward_out(Tensor & output, const Tensor & self, Scalar lambd) const override; | |
virtual Tensor softshrink_forward(const Tensor & self, Scalar lambd) const override; | |
virtual Tensor & softshrink_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, Scalar lambd) const override; | |
virtual Tensor softshrink_backward(const Tensor & grad_output, const Tensor & self, Scalar lambd) const override; | |
virtual Tensor & threshold_out(Tensor & output, const Tensor & self, Scalar threshold, Scalar value) const override; | |
virtual Tensor threshold(const Tensor & self, Scalar threshold, Scalar value) const override; | |
virtual Tensor & threshold_forward_out(Tensor & output, const Tensor & self, Scalar threshold, Scalar value) const override; | |
virtual Tensor threshold_forward(const Tensor & self, Scalar threshold, Scalar value) const override; | |
virtual Tensor & threshold_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, Scalar threshold, Scalar value) const override; | |
virtual Tensor threshold_backward(const Tensor & grad_output, const Tensor & self, Scalar threshold, Scalar value) const override; | |
virtual Tensor & threshold_(Tensor & self, Scalar threshold, Scalar value) const override; | |
virtual Tensor & threshold_forward_(Tensor & self, Scalar threshold, Scalar value) const override; | |
virtual Tensor & adaptive_avg_pool2d_out(Tensor & output, const Tensor & self, IntList output_size) const override; | |
virtual Tensor adaptive_avg_pool2d(const Tensor & self, IntList output_size) const override; | |
virtual Tensor & adaptive_avg_pool2d_forward_out(Tensor & output, const Tensor & self, IntList output_size) const override; | |
virtual Tensor adaptive_avg_pool2d_forward(const Tensor & self, IntList output_size) const override; | |
virtual Tensor & adaptive_avg_pool2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self) const override; | |
virtual Tensor adaptive_avg_pool2d_backward(const Tensor & grad_output, const Tensor & self) const override; | |
virtual Tensor & adaptive_avg_pool3d_out(Tensor & output, const Tensor & self, IntList output_size) const override; | |
virtual Tensor adaptive_avg_pool3d(const Tensor & self, IntList output_size) const override; | |
virtual Tensor & adaptive_avg_pool3d_forward_out(Tensor & output, const Tensor & self, IntList output_size) const override; | |
virtual Tensor adaptive_avg_pool3d_forward(const Tensor & self, IntList output_size) const override; | |
virtual Tensor & adaptive_avg_pool3d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self) const override; | |
virtual Tensor adaptive_avg_pool3d_backward(const Tensor & grad_output, const Tensor & self) const override; | |
virtual std::tuple<Tensor &,Tensor &> adaptive_max_pool2d_out(Tensor & output, Tensor & indices, const Tensor & self, IntList output_size) const override; | |
virtual std::tuple<Tensor,Tensor> adaptive_max_pool2d(const Tensor & self, IntList output_size) const override; | |
virtual std::tuple<Tensor &,Tensor &> adaptive_max_pool2d_forward_out(Tensor & output, Tensor & indices, const Tensor & self, IntList output_size) const override; | |
virtual std::tuple<Tensor,Tensor> adaptive_max_pool2d_forward(const Tensor & self, IntList output_size) const override; | |
virtual Tensor & adaptive_max_pool2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & indices) const override; | |
virtual Tensor adaptive_max_pool2d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & indices) const override; | |
virtual std::tuple<Tensor &,Tensor &> adaptive_max_pool3d_out(Tensor & output, Tensor & indices, const Tensor & self, IntList output_size) const override; | |
virtual std::tuple<Tensor,Tensor> adaptive_max_pool3d(const Tensor & self, IntList output_size) const override; | |
virtual std::tuple<Tensor &,Tensor &> adaptive_max_pool3d_forward_out(Tensor & output, Tensor & indices, const Tensor & self, IntList output_size) const override; | |
virtual std::tuple<Tensor,Tensor> adaptive_max_pool3d_forward(const Tensor & self, IntList output_size) const override; | |
virtual Tensor & adaptive_max_pool3d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & indices) const override; | |
virtual Tensor adaptive_max_pool3d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & indices) const override; | |
virtual Tensor & avg_pool2d_out(Tensor & output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor avg_pool2d(const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor & avg_pool2d_forward_out(Tensor & output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor avg_pool2d_forward(const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor & avg_pool2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor avg_pool2d_backward(const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor & avg_pool3d_out(Tensor & output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor avg_pool3d(const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor & avg_pool3d_forward_out(Tensor & output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor avg_pool3d_forward(const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor & avg_pool3d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual Tensor avg_pool3d_backward(const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, bool ceil_mode, bool count_include_pad) const override; | |
virtual std::tuple<Tensor &,Tensor &> fractional_max_pool2d_out(Tensor & output, Tensor & indices, const Tensor & self, IntList kernel_size, IntList output_size, const Tensor & random_samples) const override; | |
virtual std::tuple<Tensor,Tensor> fractional_max_pool2d(const Tensor & self, IntList kernel_size, IntList output_size, const Tensor & random_samples) const override; | |
virtual std::tuple<Tensor &,Tensor &> fractional_max_pool2d_forward_out(Tensor & output, Tensor & indices, const Tensor & self, IntList kernel_size, IntList output_size, const Tensor & random_samples) const override; | |
virtual std::tuple<Tensor,Tensor> fractional_max_pool2d_forward(const Tensor & self, IntList kernel_size, IntList output_size, const Tensor & random_samples) const override; | |
virtual Tensor & fractional_max_pool2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList output_size, const Tensor & indices) const override; | |
virtual Tensor fractional_max_pool2d_backward(const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList output_size, const Tensor & indices) const override; | |
virtual std::tuple<Tensor &,Tensor &> max_pool2d_out(Tensor & output, Tensor & indices, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode) const override; | |
virtual std::tuple<Tensor,Tensor> max_pool2d(const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode) const override; | |
virtual std::tuple<Tensor &,Tensor &> max_pool2d_forward_out(Tensor & output, Tensor & indices, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode) const override; | |
virtual std::tuple<Tensor,Tensor> max_pool2d_forward(const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode) const override; | |
virtual Tensor & max_pool2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode, const Tensor & indices) const override; | |
virtual Tensor max_pool2d_backward(const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode, const Tensor & indices) const override; | |
virtual std::tuple<Tensor &,Tensor &> max_pool3d_out(Tensor & output, Tensor & indices, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode) const override; | |
virtual std::tuple<Tensor,Tensor> max_pool3d(const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode) const override; | |
virtual std::tuple<Tensor &,Tensor &> max_pool3d_forward_out(Tensor & output, Tensor & indices, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode) const override; | |
virtual std::tuple<Tensor,Tensor> max_pool3d_forward(const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode) const override; | |
virtual Tensor & max_pool3d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode, const Tensor & indices) const override; | |
virtual Tensor max_pool3d_backward(const Tensor & grad_output, const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode, const Tensor & indices) const override; | |
virtual Tensor & max_unpool2d_out(Tensor & output, const Tensor & self, const Tensor & indices, IntList output_size) const override; | |
virtual Tensor max_unpool2d(const Tensor & self, const Tensor & indices, IntList output_size) const override; | |
virtual Tensor & max_unpool2d_forward_out(Tensor & output, const Tensor & self, const Tensor & indices, IntList output_size) const override; | |
virtual Tensor max_unpool2d_forward(const Tensor & self, const Tensor & indices, IntList output_size) const override; | |
virtual Tensor & max_unpool2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & indices, IntList output_size) const override; | |
virtual Tensor max_unpool2d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & indices, IntList output_size) const override; | |
virtual Tensor & max_unpool3d_out(Tensor & output, const Tensor & self, const Tensor & indices, IntList output_size, IntList stride, IntList padding) const override; | |
virtual Tensor max_unpool3d(const Tensor & self, const Tensor & indices, IntList output_size, IntList stride, IntList padding) const override; | |
virtual Tensor & max_unpool3d_forward_out(Tensor & output, const Tensor & self, const Tensor & indices, IntList output_size, IntList stride, IntList padding) const override; | |
virtual Tensor max_unpool3d_forward(const Tensor & self, const Tensor & indices, IntList output_size, IntList stride, IntList padding) const override; | |
virtual Tensor & max_unpool3d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, const Tensor & indices, IntList output_size, IntList stride, IntList padding) const override; | |
virtual Tensor max_unpool3d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & indices, IntList output_size, IntList stride, IntList padding) const override; | |
virtual Tensor & reflection_pad1d_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor reflection_pad1d(const Tensor & self, IntList padding) const override; | |
virtual Tensor & reflection_pad1d_forward_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor reflection_pad1d_forward(const Tensor & self, IntList padding) const override; | |
virtual Tensor & reflection_pad1d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor reflection_pad1d_backward(const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor & reflection_pad2d_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor reflection_pad2d(const Tensor & self, IntList padding) const override; | |
virtual Tensor & reflection_pad2d_forward_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor reflection_pad2d_forward(const Tensor & self, IntList padding) const override; | |
virtual Tensor & reflection_pad2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor reflection_pad2d_backward(const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor & replication_pad1d_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor replication_pad1d(const Tensor & self, IntList padding) const override; | |
virtual Tensor & replication_pad1d_forward_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor replication_pad1d_forward(const Tensor & self, IntList padding) const override; | |
virtual Tensor & replication_pad1d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor replication_pad1d_backward(const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor & replication_pad2d_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor replication_pad2d(const Tensor & self, IntList padding) const override; | |
virtual Tensor & replication_pad2d_forward_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor replication_pad2d_forward(const Tensor & self, IntList padding) const override; | |
virtual Tensor & replication_pad2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor replication_pad2d_backward(const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor & replication_pad3d_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor replication_pad3d(const Tensor & self, IntList padding) const override; | |
virtual Tensor & replication_pad3d_forward_out(Tensor & output, const Tensor & self, IntList padding) const override; | |
virtual Tensor replication_pad3d_forward(const Tensor & self, IntList padding) const override; | |
virtual Tensor & replication_pad3d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor replication_pad3d_backward(const Tensor & grad_output, const Tensor & self, IntList padding) const override; | |
virtual Tensor & upsample_linear1d_out(Tensor & output, const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor upsample_linear1d(const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor & upsample_linear1d_forward_out(Tensor & output, const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor upsample_linear1d_forward(const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor & upsample_linear1d_backward_out(Tensor & grad_input, const Tensor & grad_output, IntList output_size, IntList input_size, bool align_corners) const override; | |
virtual Tensor upsample_linear1d_backward(const Tensor & grad_output, IntList output_size, IntList input_size, bool align_corners) const override; | |
virtual Tensor & upsample_bilinear2d_out(Tensor & output, const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor upsample_bilinear2d(const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor & upsample_bilinear2d_forward_out(Tensor & output, const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor upsample_bilinear2d_forward(const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor & upsample_bilinear2d_backward_out(Tensor & grad_input, const Tensor & grad_output, IntList output_size, IntList input_size, bool align_corners) const override; | |
virtual Tensor upsample_bilinear2d_backward(const Tensor & grad_output, IntList output_size, IntList input_size, bool align_corners) const override; | |
virtual Tensor & upsample_trilinear3d_out(Tensor & output, const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor upsample_trilinear3d(const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor & upsample_trilinear3d_forward_out(Tensor & output, const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor upsample_trilinear3d_forward(const Tensor & self, IntList output_size, bool align_corners) const override; | |
virtual Tensor & upsample_trilinear3d_backward_out(Tensor & grad_input, const Tensor & grad_output, IntList output_size, IntList input_size, bool align_corners) const override; | |
virtual Tensor upsample_trilinear3d_backward(const Tensor & grad_output, IntList output_size, IntList input_size, bool align_corners) const override; | |
virtual Tensor & upsample_nearest1d_out(Tensor & output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor upsample_nearest1d(const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor & upsample_nearest1d_forward_out(Tensor & output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor upsample_nearest1d_forward(const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor & upsample_nearest1d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor upsample_nearest1d_backward(const Tensor & grad_output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor & upsample_nearest2d_out(Tensor & output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor upsample_nearest2d(const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor & upsample_nearest2d_forward_out(Tensor & output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor upsample_nearest2d_forward(const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor & upsample_nearest2d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor upsample_nearest2d_backward(const Tensor & grad_output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor & upsample_nearest3d_out(Tensor & output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor upsample_nearest3d(const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor & upsample_nearest3d_forward_out(Tensor & output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor upsample_nearest3d_forward(const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor & upsample_nearest3d_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor upsample_nearest3d_backward(const Tensor & grad_output, const Tensor & self, int64_t scale_factor) const override; | |
virtual Tensor & _sigmoid_out(Tensor & output, const Tensor & self) const override; | |
virtual Tensor _sigmoid(const Tensor & self) const override; | |
virtual Tensor & _sigmoid_forward_out(Tensor & output, const Tensor & self) const override; | |
virtual Tensor _sigmoid_forward(const Tensor & self) const override; | |
virtual Tensor & _sigmoid_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & output) const override; | |
virtual Tensor _sigmoid_backward(const Tensor & grad_output, const Tensor & output) const override; | |
virtual Tensor & _tanh_out(Tensor & output, const Tensor & self) const override; | |
virtual Tensor _tanh(const Tensor & self) const override; | |
virtual Tensor & _tanh_forward_out(Tensor & output, const Tensor & self) const override; | |
virtual Tensor _tanh_forward(const Tensor & self) const override; | |
virtual Tensor & _tanh_backward_out(Tensor & grad_input, const Tensor & grad_output, const Tensor & output) const override; | |
virtual Tensor _tanh_backward(const Tensor & grad_output, const Tensor & output) const override; | |
virtual Tensor & thnn_batch_norm_out(Tensor & output, const Tensor & self, const Tensor & weight, const Tensor & bias, const Tensor & running_mean, const Tensor & running_var, bool training, double momentum, double eps) const override; | |
virtual Tensor thnn_batch_norm(const Tensor & self, const Tensor & weight, const Tensor & bias, const Tensor & running_mean, const Tensor & running_var, bool training, double momentum, double eps) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_batch_norm_forward_out(Tensor & output, Tensor & save_mean, Tensor & save_std, const Tensor & self, const Tensor & weight, const Tensor & bias, const Tensor & running_mean, const Tensor & running_var, bool training, double momentum, double eps) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_batch_norm_forward(const Tensor & self, const Tensor & weight, const Tensor & bias, const Tensor & running_mean, const Tensor & running_var, bool training, double momentum, double eps) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_batch_norm_backward_out(Tensor & grad_input, Tensor & grad_weight, Tensor & grad_bias, const Tensor & grad_output, const Tensor & self, const Tensor & weight, const Tensor & running_mean, const Tensor & running_var, bool training, double eps, const Tensor & save_mean, const Tensor & save_std) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_batch_norm_backward(const Tensor & grad_output, const Tensor & self, const Tensor & weight, const Tensor & running_mean, const Tensor & running_var, bool training, double eps, const Tensor & save_mean, const Tensor & save_std, std::array<bool,3> output_mask) const override; | |
virtual Tensor & thnn_conv_transpose2d_out(Tensor & output, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, IntList dilation) const override; | |
virtual Tensor thnn_conv_transpose2d(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, IntList dilation) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv_transpose2d_forward_out(Tensor & output, Tensor & columns, Tensor & ones, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, IntList dilation) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv_transpose2d_forward(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, IntList dilation) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv_transpose2d_backward_out(Tensor & grad_input, Tensor & grad_weight, Tensor & grad_bias, const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList output_padding, IntList dilation, const Tensor & columns, const Tensor & ones) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv_transpose2d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList output_padding, IntList dilation, const Tensor & columns, const Tensor & ones, std::array<bool,3> output_mask) const override; | |
virtual Tensor & thnn_conv_transpose3d_out(Tensor & output, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, IntList dilation) const override; | |
virtual Tensor thnn_conv_transpose3d(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, IntList dilation) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv_transpose3d_forward_out(Tensor & output, Tensor & finput, Tensor & fgrad_input, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, IntList dilation) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv_transpose3d_forward(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, IntList dilation) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv_transpose3d_backward_out(Tensor & grad_input, Tensor & grad_weight, Tensor & grad_bias, const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList output_padding, IntList dilation, const Tensor & finput, const Tensor & fgrad_input) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv_transpose3d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList output_padding, IntList dilation, const Tensor & finput, const Tensor & fgrad_input, std::array<bool,3> output_mask) const override; | |
virtual Tensor & thnn_conv2d_out(Tensor & output, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding) const override; | |
virtual Tensor thnn_conv2d(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv2d_forward_out(Tensor & output, Tensor & finput, Tensor & fgrad_input, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv2d_forward(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv2d_backward_out(Tensor & grad_input, Tensor & grad_weight, Tensor & grad_bias, const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, const Tensor & finput, const Tensor & fgrad_input) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv2d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, const Tensor & finput, const Tensor & fgrad_input, std::array<bool,3> output_mask) const override; | |
virtual Tensor & thnn_conv_depthwise2d_out(Tensor & output, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual Tensor thnn_conv_depthwise2d(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual Tensor & thnn_conv_depthwise2d_forward_out(Tensor & output, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual Tensor thnn_conv_depthwise2d_forward(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual std::tuple<Tensor &,Tensor &> thnn_conv_depthwise2d_backward_out(Tensor & grad_input, Tensor & grad_weight, const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList dilation) const override; | |
virtual std::tuple<Tensor,Tensor> thnn_conv_depthwise2d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList dilation, std::array<bool,2> output_mask) const override; | |
virtual Tensor & thnn_conv3d_out(Tensor & output, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding) const override; | |
virtual Tensor thnn_conv3d(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv3d_forward_out(Tensor & output, Tensor & finput, Tensor & fgrad_input, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv3d_forward(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv3d_backward_out(Tensor & grad_input, Tensor & grad_weight, Tensor & grad_bias, const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, const Tensor & finput, const Tensor & fgrad_input) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv3d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, const Tensor & finput, const Tensor & fgrad_input, std::array<bool,3> output_mask) const override; | |
virtual Tensor & thnn_conv_dilated2d_out(Tensor & output, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual Tensor thnn_conv_dilated2d(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv_dilated2d_forward_out(Tensor & output, Tensor & columns, Tensor & ones, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv_dilated2d_forward(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv_dilated2d_backward_out(Tensor & grad_input, Tensor & grad_weight, Tensor & grad_bias, const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList dilation, const Tensor & columns, const Tensor & ones) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv_dilated2d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList dilation, const Tensor & columns, const Tensor & ones, std::array<bool,3> output_mask) const override; | |
virtual Tensor & thnn_conv_dilated3d_out(Tensor & output, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual Tensor thnn_conv_dilated3d(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv_dilated3d_forward_out(Tensor & output, Tensor & columns, Tensor & ones, const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv_dilated3d_forward(const Tensor & self, const Tensor & weight, IntList kernel_size, const Tensor & bias, IntList stride, IntList padding, IntList dilation) const override; | |
virtual std::tuple<Tensor &,Tensor &,Tensor &> thnn_conv_dilated3d_backward_out(Tensor & grad_input, Tensor & grad_weight, Tensor & grad_bias, const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList dilation, const Tensor & columns, const Tensor & ones) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> thnn_conv_dilated3d_backward(const Tensor & grad_output, const Tensor & self, const Tensor & weight, IntList kernel_size, IntList stride, IntList padding, IntList dilation, const Tensor & columns, const Tensor & ones, std::array<bool,3> output_mask) const override; | |
virtual Tensor _cast_Byte(const Tensor & self, bool non_blocking) const override; | |
virtual Tensor _cast_Char(const Tensor & self, bool non_blocking) const override; | |
virtual Tensor _cast_Double(const Tensor & self, bool non_blocking) const override; | |
virtual Tensor _cast_Float(const Tensor & self, bool non_blocking) const override; | |
virtual Tensor _cast_Int(const Tensor & self, bool non_blocking) const override; | |
virtual Tensor _cast_Long(const Tensor & self, bool non_blocking) const override; | |
virtual Tensor _cast_Short(const Tensor & self, bool non_blocking) const override; | |
virtual Tensor _cast_Half(const Tensor & self, bool non_blocking) const override; | |
virtual Tensor _cudnn_rnn_flatten_weight(TensorList weight_arr, int64_t weight_stride0, int64_t input_size, int64_t mode, int64_t hidden_size, int64_t num_layers, bool batch_first, bool bidirectional) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor,Tensor,Tensor> _cudnn_rnn(const Tensor & input, TensorList weight, int64_t weight_stride0, const Tensor & weight_buf, const Tensor & hx, const Tensor & cx, int64_t mode, int64_t hidden_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, IntList batch_sizes, const Tensor & dropout_state) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor,std::vector<Tensor>> _cudnn_rnn_backward(const Tensor & input, TensorList weight, int64_t weight_stride0, const Tensor & weight_buf, const Tensor & hx, const Tensor & cx, const Tensor & output, const Tensor & grad_output, const Tensor & grad_hy, const Tensor & grad_cy, int64_t mode, int64_t hidden_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, IntList batch_sizes, const Tensor & dropout_state, const Tensor & reserve, std::array<bool,4> output_mask) const override; | |
virtual Tensor _cudnn_init_dropout_state(double dropout, bool train, int64_t dropout_seed) const override; | |
virtual Tensor abs(const Tensor & self) const override; | |
virtual Tensor & abs_(Tensor & self) const override; | |
virtual Tensor & abs_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor acos(const Tensor & self) const override; | |
virtual Tensor & acos_(Tensor & self) const override; | |
virtual Tensor & acos_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor adaptive_avg_pool1d(const Tensor & self, IntList output_size) const override; | |
virtual std::tuple<Tensor,Tensor> adaptive_max_pool1d(const Tensor & self, IntList output_size) const override; | |
virtual bool allclose(const Tensor & self, const Tensor & other, double rtol, double atol, bool equal_nan) const override; | |
virtual Tensor addmv(const Tensor & self, const Tensor & mat, const Tensor & vec, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & addmv_(Tensor & self, const Tensor & mat, const Tensor & vec, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & addmv_out(Tensor & result, const Tensor & self, const Tensor & mat, const Tensor & vec, Scalar beta, Scalar alpha) const override; | |
virtual Tensor addr(const Tensor & self, const Tensor & vec1, const Tensor & vec2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & addr_(Tensor & self, const Tensor & vec1, const Tensor & vec2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & addr_out(Tensor & result, const Tensor & self, const Tensor & vec1, const Tensor & vec2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor arange(Scalar start, Scalar end, Scalar step) const override; | |
virtual Tensor & arange_out(Tensor & result, Scalar start, Scalar end, Scalar step) const override; | |
virtual Tensor arange(Scalar end) const override; | |
virtual Tensor & arange_out(Tensor & result, Scalar end) const override; | |
virtual Tensor argmax(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor argmax(const Tensor & self) const override; | |
virtual Tensor _argmax(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor argmin(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor argmin(const Tensor & self) const override; | |
virtual Tensor _argmin(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor asin(const Tensor & self) const override; | |
virtual Tensor & asin_(Tensor & self) const override; | |
virtual Tensor & asin_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor atan(const Tensor & self) const override; | |
virtual Tensor & atan_(Tensor & self) const override; | |
virtual Tensor & atan_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor bartlett_window(int64_t window_length, bool periodic) const override; | |
virtual Tensor batch_norm(const Tensor & input, const Tensor & weight, const Tensor & bias, const Tensor & running_mean, const Tensor & running_var, bool training, double momentum, double eps, bool cudnn_enabled) const override; | |
virtual Tensor bernoulli(const Tensor & self, const Tensor & p, Generator * generator) const override; | |
virtual Tensor bernoulli(const Tensor & self, double p, Generator * generator) const override; | |
virtual Tensor bernoulli(const Tensor & self) const override; | |
virtual Tensor & bernoulli_(Tensor & self, const Tensor & p, Generator * generator) const override; | |
virtual Tensor & bernoulli_(Tensor & self, double p, Generator * generator) const override; | |
virtual Tensor & bernoulli_(Tensor & self) const override; | |
virtual Tensor bilinear(const Tensor & input1, const Tensor & input2, const Tensor & weight, const Tensor & bias) const override; | |
virtual Tensor blackman_window(int64_t window_length, bool periodic) const override; | |
virtual Tensor cat(TensorList tensors, int64_t dim) const override; | |
virtual Tensor & cat_out(Tensor & result, TensorList tensors, int64_t dim) const override; | |
virtual Tensor ceil(const Tensor & self) const override; | |
virtual Tensor & ceil_(Tensor & self) const override; | |
virtual Tensor & ceil_out(Tensor & result, const Tensor & self) const override; | |
virtual std::vector<Tensor> chunk(const Tensor & self, int64_t chunks, int64_t dim) const override; | |
virtual bool cudnn_is_acceptable(const Tensor & self) const override; | |
virtual Tensor convolution(const Tensor & input, const Tensor & weight, const Tensor & bias, IntList stride, IntList padding, IntList dilation, bool transposed, IntList output_padding, int64_t groups) const override; | |
virtual Tensor _convolution(const Tensor & input, const Tensor & weight, const Tensor & bias, IntList stride, IntList padding, IntList dilation, bool transposed, IntList output_padding, int64_t groups, bool benchmark, bool deterministic, bool cudnn_enabled) const override; | |
virtual Tensor _convolution_nogroup(const Tensor & input, const Tensor & weight, const Tensor & bias, IntList stride, IntList padding, IntList dilation, bool transposed, IntList output_padding) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> _convolution_double_backward(const Tensor & ggI, const Tensor & ggW, const Tensor & ggb, const Tensor & gO, const Tensor & weight, const Tensor & self, IntList stride, IntList padding, IntList dilation, bool transposed, IntList output_padding, int64_t groups, bool benchmark, bool deterministic, bool cudnn_enabled, std::array<bool,3> output_mask) const override; | |
virtual Tensor conv1d(const Tensor & input, const Tensor & weight, const Tensor & bias, IntList stride, IntList padding, IntList dilation, int64_t groups) const override; | |
virtual Tensor conv2d(const Tensor & input, const Tensor & weight, const Tensor & bias, IntList stride, IntList padding, IntList dilation, int64_t groups) const override; | |
virtual Tensor conv3d(const Tensor & input, const Tensor & weight, const Tensor & bias, IntList stride, IntList padding, IntList dilation, int64_t groups) const override; | |
virtual Tensor conv_tbc(const Tensor & self, const Tensor & weight, const Tensor & bias, int64_t pad) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> conv_tbc_backward(const Tensor & self, const Tensor & input, const Tensor & weight, const Tensor & bias, int64_t pad) const override; | |
virtual Tensor conv_transpose1d(const Tensor & input, const Tensor & weight, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, int64_t groups, IntList dilation) const override; | |
virtual Tensor conv_transpose2d(const Tensor & input, const Tensor & weight, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, int64_t groups, IntList dilation) const override; | |
virtual Tensor conv_transpose3d(const Tensor & input, const Tensor & weight, const Tensor & bias, IntList stride, IntList padding, IntList output_padding, int64_t groups, IntList dilation) const override; | |
virtual Tensor cos(const Tensor & self) const override; | |
virtual Tensor & cos_(Tensor & self) const override; | |
virtual Tensor & cos_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor cosh(const Tensor & self) const override; | |
virtual Tensor & cosh_(Tensor & self) const override; | |
virtual Tensor & cosh_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor cosine_embedding_loss(const Tensor & input1, const Tensor & input2, const Tensor & target, double margin, bool size_average, bool reduce) const override; | |
virtual Tensor cudnn_affine_grid_generator(const Tensor & theta, int64_t N, int64_t C, int64_t H, int64_t W) const override; | |
virtual Tensor cudnn_affine_grid_generator_backward(const Tensor & grad, int64_t N, int64_t C, int64_t H, int64_t W) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> cudnn_batch_norm(const Tensor & input, const Tensor & weight, const Tensor & bias, const Tensor & running_mean, const Tensor & running_var, bool training, double exponential_average_factor, double epsilon) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> cudnn_batch_norm_backward(const Tensor & input, const Tensor & grad_output, const Tensor & weight, const Tensor & running_mean, const Tensor & running_var, const Tensor & save_mean, const Tensor & save_var, double epsilon) const override; | |
virtual Tensor cudnn_convolution(const Tensor & self, const Tensor & weight, const Tensor & bias, IntList padding, IntList stride, IntList dilation, int64_t groups, bool benchmark, bool deterministic) const override; | |
virtual Tensor cudnn_convolution_backward_input(IntList self_size, const Tensor & grad_output, const Tensor & weight, IntList padding, IntList stride, IntList dilation, int64_t groups, bool benchmark, bool deterministic) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> cudnn_convolution_backward(const Tensor & self, const Tensor & grad_output, const Tensor & weight, IntList padding, IntList stride, IntList dilation, int64_t groups, bool benchmark, bool deterministic, std::array<bool,3> output_mask) const override; | |
virtual Tensor cudnn_convolution_backward_bias(const Tensor & grad_output) const override; | |
virtual Tensor cudnn_convolution_backward_weight(IntList weight_size, const Tensor & grad_output, const Tensor & self, IntList padding, IntList stride, IntList dilation, int64_t groups, bool benchmark, bool deterministic) const override; | |
virtual Tensor cudnn_convolution_transpose(const Tensor & self, const Tensor & weight, const Tensor & bias, IntList padding, IntList output_padding, IntList stride, IntList dilation, int64_t groups, bool benchmark, bool deterministic) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> cudnn_convolution_transpose_backward(const Tensor & self, const Tensor & grad_output, const Tensor & weight, IntList padding, IntList output_padding, IntList stride, IntList dilation, int64_t groups, bool benchmark, bool deterministic, std::array<bool,3> output_mask) const override; | |
virtual Tensor cudnn_convolution_transpose_backward_bias(const Tensor & grad_output) const override; | |
virtual Tensor cudnn_convolution_transpose_backward_input(const Tensor & grad_output, const Tensor & weight, IntList padding, IntList stride, IntList dilation, int64_t groups, bool benchmark, bool deterministic) const override; | |
virtual Tensor cudnn_convolution_transpose_backward_weight(IntList weight_size, const Tensor & grad_output, const Tensor & self, IntList padding, IntList stride, IntList dilation, int64_t groups, bool benchmark, bool deterministic) const override; | |
virtual Tensor cudnn_grid_sampler(const Tensor & self, const Tensor & grid) const override; | |
virtual std::tuple<Tensor,Tensor> cudnn_grid_sampler_backward(const Tensor & self, const Tensor & grid, const Tensor & grad_output) const override; | |
virtual Tensor cumsum(const Tensor & self, int64_t dim, ScalarType dtype) const override; | |
virtual Tensor cumsum(const Tensor & self, int64_t dim) const override; | |
virtual Tensor & cumsum_out(Tensor & result, const Tensor & self, int64_t dim, ScalarType dtype) const override; | |
virtual Tensor & cumsum_out(Tensor & result, const Tensor & self, int64_t dim) const override; | |
virtual Tensor cumprod(const Tensor & self, int64_t dim, ScalarType dtype) const override; | |
virtual Tensor cumprod(const Tensor & self, int64_t dim) const override; | |
virtual Tensor & cumprod_out(Tensor & result, const Tensor & self, int64_t dim, ScalarType dtype) const override; | |
virtual Tensor & cumprod_out(Tensor & result, const Tensor & self, int64_t dim) const override; | |
virtual Tensor det(const Tensor & self) const override; | |
virtual Tensor diagflat(const Tensor & self, int64_t offset) const override; | |
virtual Tensor diagonal(const Tensor & self, int64_t offset, int64_t dim1, int64_t dim2) const override; | |
virtual Tensor dot(const Tensor & self, const Tensor & tensor) const override; | |
virtual Tensor & dot_out(Tensor & result, const Tensor & self, const Tensor & tensor) const override; | |
virtual Tensor einsum(std::string equation, TensorList tensors) const override; | |
virtual Tensor embedding(const Tensor & weight, const Tensor & indices, int64_t padding_idx, bool scale_grad_by_freq, bool sparse) const override; | |
virtual Tensor embedding_backward(const Tensor & grad, const Tensor & indices, int64_t num_weights, int64_t padding_idx, bool scale_grad_by_freq, bool sparse) const override; | |
virtual Tensor embedding_dense_backward(const Tensor & grad, const Tensor & indices, int64_t num_weights, int64_t padding_idx, bool scale_grad_by_freq) const override; | |
virtual Tensor & embedding_renorm_(Tensor & self, const Tensor & indices, double max_norm, double norm_type) const override; | |
virtual Tensor embedding_sparse_backward(const Tensor & grad, const Tensor & indices, int64_t num_weights, int64_t padding_idx, bool scale_grad_by_freq) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor,Tensor> embedding_bag(const Tensor & weight, const Tensor & indices, const Tensor & offsets, bool scale_grad_by_freq, int64_t mode, bool sparse) const override; | |
virtual Tensor embedding_bag_backward(const Tensor & grad, const Tensor & indices, const Tensor & offsets, const Tensor & offset2bag, const Tensor & bag_size, const Tensor & maximum_indices, int64_t num_weights, bool scale_grad_by_freq, int64_t mode, bool sparse) const override; | |
virtual Tensor embedding_bag_sparse_backward(const Tensor & grad, const Tensor & indices, const Tensor & offsets, const Tensor & offset2bag, const Tensor & bag_size, int64_t num_weights, bool scale_grad_by_freq, int64_t mode) const override; | |
virtual Tensor embedding_bag_dense_backward(const Tensor & grad, const Tensor & indices, const Tensor & offsets, const Tensor & offset2bag, const Tensor & bag_size, const Tensor & maximum_indices, int64_t num_weights, bool scale_grad_by_freq, int64_t mode) const override; | |
virtual Tensor empty(IntList size) const override; | |
virtual Tensor & empty_out(Tensor & result, IntList size) const override; | |
virtual Tensor empty_like(const Tensor & self) const override; | |
virtual Tensor empty_like(const Tensor & self, const Type & dtype) const override; | |
virtual Tensor erf(const Tensor & self) const override; | |
virtual Tensor & erf_(Tensor & self) const override; | |
virtual Tensor & erf_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor exp(const Tensor & self) const override; | |
virtual Tensor & exp_(Tensor & self) const override; | |
virtual Tensor & exp_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor expm1(const Tensor & self) const override; | |
virtual Tensor & expm1_(Tensor & self) const override; | |
virtual Tensor & expm1_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor expand(const Tensor & self, IntList size, bool implicit) const override; | |
virtual Tensor expand_as(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor eye(int64_t n, int64_t m) const override; | |
virtual Tensor & eye_out(Tensor & result, int64_t n, int64_t m) const override; | |
virtual Tensor & fill_(Tensor & self, Scalar value) const override; | |
virtual Tensor & fill_(Tensor & self, const Tensor & value) const override; | |
virtual Tensor floor(const Tensor & self) const override; | |
virtual Tensor & floor_(Tensor & self) const override; | |
virtual Tensor & floor_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor full(IntList size, Scalar fill_value) const override; | |
virtual Tensor & full_out(Tensor & result, IntList size, Scalar fill_value) const override; | |
virtual Tensor full_like(const Tensor & self, Scalar fill_value) const override; | |
virtual Tensor full_like(const Tensor & self, Scalar fill_value, const Type & dtype) const override; | |
virtual Tensor hann_window(int64_t window_length, bool periodic) const override; | |
virtual Tensor hamming_window(int64_t window_length, bool periodic, double alpha, double beta) const override; | |
virtual Tensor hinge_embedding_loss(const Tensor & self, const Tensor & target, double margin, bool size_average, bool reduce) const override; | |
virtual Tensor ger(const Tensor & self, const Tensor & vec2) const override; | |
virtual Tensor & ger_out(Tensor & result, const Tensor & self, const Tensor & vec2) const override; | |
virtual std::tuple<Tensor,Tensor> gesv(const Tensor & self, const Tensor & A) const override; | |
virtual std::tuple<Tensor &,Tensor &> gesv_out(Tensor & solution, Tensor & lu, const Tensor & self, const Tensor & A) const override; | |
virtual std::tuple<Tensor,Tensor> _gesv_helper(const Tensor & self, const Tensor & A) const override; | |
virtual Tensor group_norm(const Tensor & input, int64_t num_groups, const Tensor & weight, const Tensor & bias, double eps, bool cudnn_enabled) const override; | |
virtual Tensor fft(const Tensor & self, int64_t signal_ndim, bool normalized) const override; | |
virtual Tensor ifft(const Tensor & self, int64_t signal_ndim, bool normalized) const override; | |
virtual Tensor rfft(const Tensor & self, int64_t signal_ndim, bool normalized, bool onesided) const override; | |
virtual Tensor irfft(const Tensor & self, int64_t signal_ndim, bool normalized, bool onesided, IntList signal_sizes) const override; | |
virtual Tensor _fft_with_size(const Tensor & self, int64_t signal_ndim, bool complex_input, bool complex_output, bool inverse, IntList checked_signal_sizes, bool normalized, bool onesided, IntList output_sizes) const override; | |
virtual Tensor index(const Tensor & self, TensorList indices) const override; | |
virtual Tensor & index_copy_(Tensor & self, int64_t dim, const Tensor & index, const Tensor & source) const override; | |
virtual Tensor index_put(const Tensor & self, TensorList indices, const Tensor & values) const override; | |
virtual Tensor & index_put_(Tensor & self, TensorList indices, const Tensor & values) const override; | |
virtual Tensor isclose(const Tensor & self, const Tensor & other, double rtol, double atol, bool equal_nan) const override; | |
virtual bool is_cuda(const Tensor & self) const override; | |
virtual bool is_distributed(const Tensor & self) const override; | |
virtual bool is_floating_point(const Tensor & self) const override; | |
virtual bool is_nonzero(const Tensor & self) const override; | |
virtual bool is_same_size(const Tensor & self, const Tensor & other) const override; | |
virtual bool is_signed(const Tensor & self) const override; | |
virtual bool is_sparse(const Tensor & self) const override; | |
virtual Tensor layer_norm(const Tensor & input, IntList normalized_shape, const Tensor & weight, const Tensor & bias, double eps, bool cudnn_enable) const override; | |
virtual Tensor linspace(Scalar start, Scalar end, int64_t steps) const override; | |
virtual Tensor & linspace_out(Tensor & result, Scalar start, Scalar end, int64_t steps) const override; | |
virtual Tensor log(const Tensor & self) const override; | |
virtual Tensor & log_(Tensor & self) const override; | |
virtual Tensor & log_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor log10(const Tensor & self) const override; | |
virtual Tensor & log10_(Tensor & self) const override; | |
virtual Tensor & log10_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor log1p(const Tensor & self) const override; | |
virtual Tensor & log1p_(Tensor & self) const override; | |
virtual Tensor & log1p_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor log2(const Tensor & self) const override; | |
virtual Tensor & log2_(Tensor & self) const override; | |
virtual Tensor & log2_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor logdet(const Tensor & self) const override; | |
virtual Tensor logspace(Scalar start, Scalar end, int64_t steps) const override; | |
virtual Tensor & logspace_out(Tensor & result, Scalar start, Scalar end, int64_t steps) const override; | |
virtual Tensor log_softmax(const Tensor & self, int64_t dim) const override; | |
virtual Tensor log_softmax_backward_data(const Tensor & grad_output, const Tensor & output, int64_t dim, const Tensor & self) const override; | |
virtual Tensor logsumexp(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor & logsumexp_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor margin_ranking_loss(const Tensor & input1, const Tensor & input2, const Tensor & target, double margin, bool size_average, bool reduce) const override; | |
virtual Tensor matmul(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor & matmul_out(Tensor & result, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor max_values(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual std::tuple<Tensor,Tensor> max_pool1d(const Tensor & self, IntList kernel_size, IntList stride, IntList padding, IntList dilation, bool ceil_mode) const override; | |
virtual Tensor min_values(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor mkldnn_convolution(const Tensor & self, const Tensor & weight, const Tensor & bias, IntList padding, IntList stride, IntList dilation) const override; | |
virtual Tensor mkldnn_convolution_backward_input(IntList self_size, const Tensor & grad_output, const Tensor & weight, IntList padding, IntList stride, IntList dilation, bool bias_defined) const override; | |
virtual std::tuple<Tensor,Tensor> mkldnn_convolution_backward_weights(IntList weight_size, const Tensor & grad_output, const Tensor & self, IntList padding, IntList stride, IntList dilation, bool bias_defined) const override; | |
virtual std::tuple<Tensor,Tensor,Tensor> mkldnn_convolution_backward(const Tensor & self, const Tensor & grad_output, const Tensor & weight, IntList padding, IntList stride, IntList dilation, std::array<bool,3> output_mask) const override; | |
virtual Tensor mm(const Tensor & self, const Tensor & mat2) const override; | |
virtual Tensor & mm_out(Tensor & result, const Tensor & self, const Tensor & mat2) const override; | |
virtual Tensor mv(const Tensor & self, const Tensor & vec) const override; | |
virtual Tensor & mv_out(Tensor & result, const Tensor & self, const Tensor & vec) const override; | |
virtual Tensor narrow(const Tensor & self, int64_t dim, int64_t start, int64_t length) const override; | |
virtual Tensor ones(IntList size) const override; | |
virtual Tensor & ones_out(Tensor & result, IntList size) const override; | |
virtual Tensor ones_like(const Tensor & self) const override; | |
virtual Tensor ones_like(const Tensor & self, const Type & dtype) const override; | |
virtual Tensor pairwise_distance(const Tensor & x1, const Tensor & x2, double p, double eps, bool keepdim) const override; | |
virtual Tensor permute(const Tensor & self, IntList dims) const override; | |
virtual Tensor pin_memory(const Tensor & self) const override; | |
virtual Tensor rand(IntList size, Generator * generator) const override; | |
virtual Tensor & rand_out(Tensor & result, IntList size, Generator * generator) const override; | |
virtual Tensor rand_like(const Tensor & self) const override; | |
virtual Tensor rand_like(const Tensor & self, const Type & dtype) const override; | |
virtual Tensor randint(int64_t high, IntList size, Generator * generator) const override; | |
virtual Tensor randint(int64_t low, int64_t high, IntList size, Generator * generator) const override; | |
virtual Tensor & randint_out(Tensor & result, int64_t high, IntList size, Generator * generator) const override; | |
virtual Tensor & randint_out(Tensor & result, int64_t low, int64_t high, IntList size, Generator * generator) const override; | |
virtual Tensor randint_like(const Tensor & self, int64_t high) const override; | |
virtual Tensor randint_like(const Tensor & self, int64_t low, int64_t high) const override; | |
virtual Tensor randint_like(const Tensor & self, int64_t high, const Type & dtype) const override; | |
virtual Tensor randint_like(const Tensor & self, int64_t low, int64_t high, const Type & dtype) const override; | |
virtual Tensor randn(IntList size, Generator * generator) const override; | |
virtual Tensor & randn_out(Tensor & result, IntList size, Generator * generator) const override; | |
virtual Tensor randn_like(const Tensor & self) const override; | |
virtual Tensor randn_like(const Tensor & self, const Type & dtype) const override; | |
virtual Tensor randperm(int64_t n, Generator * generator) const override; | |
virtual Tensor & randperm_out(Tensor & result, int64_t n, Generator * generator) const override; | |
virtual Tensor range(Scalar start, Scalar end, Scalar step) const override; | |
virtual Tensor & range_out(Tensor & result, Scalar start, Scalar end, Scalar step) const override; | |
virtual Tensor repeat(const Tensor & self, IntList repeats) const override; | |
virtual Tensor reshape(const Tensor & self, IntList shape) const override; | |
virtual std::tuple<Tensor,Tensor> RoiPooling2d_forward(const Tensor & input, const Tensor & rois, int64_t pooledHeight, int64_t pooledWidth, double spatialScale) const override; | |
virtual Tensor RoiPooling2d_backward(const Tensor & input, const Tensor & rois, int64_t pooledHeight, int64_t pooledWidth, double spatialScale, const Tensor & gradOutput, const Tensor & argmaxes) const override; | |
virtual Tensor round(const Tensor & self) const override; | |
virtual Tensor & round_(Tensor & self) const override; | |
virtual Tensor & round_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor rrelu(const Tensor & self, Scalar lower, Scalar upper, bool training, Generator * generator) const override; | |
virtual Tensor & rrelu_(Tensor & self, Scalar lower, Scalar upper, bool training, Generator * generator) const override; | |
virtual Tensor relu(const Tensor & self) const override; | |
virtual Tensor & relu_(Tensor & self) const override; | |
virtual Tensor rsqrt(const Tensor & self) const override; | |
virtual Tensor & rsqrt_(Tensor & self) const override; | |
virtual Tensor & rsqrt_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor select(const Tensor & self, int64_t dim, int64_t index) const override; | |
virtual Tensor selu(const Tensor & self) const override; | |
virtual Tensor & selu_(Tensor & self) const override; | |
virtual Tensor sin(const Tensor & self) const override; | |
virtual Tensor & sin_(Tensor & self) const override; | |
virtual Tensor & sin_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor sinh(const Tensor & self) const override; | |
virtual Tensor & sinh_(Tensor & self) const override; | |
virtual Tensor & sinh_out(Tensor & result, const Tensor & self) const override; | |
virtual int64_t size(const Tensor & self, int64_t dim) const override; | |
virtual Tensor slice(const Tensor & self, int64_t dim, int64_t start, int64_t end, int64_t step) const override; | |
virtual std::tuple<Tensor,Tensor> slogdet(const Tensor & self) const override; | |
virtual Tensor smm(const Tensor & self, const Tensor & mat2) const override; | |
virtual Tensor softmax(const Tensor & self, int64_t dim) const override; | |
virtual Tensor softmax_backward_data(const Tensor & grad_output, const Tensor & output, int64_t dim, const Tensor & self) const override; | |
virtual std::vector<Tensor> split(const Tensor & self, int64_t split_size, int64_t dim) const override; | |
virtual std::vector<Tensor> split_with_sizes(const Tensor & self, IntList split_sizes, int64_t dim) const override; | |
virtual Tensor squeeze(const Tensor & self) const override; | |
virtual Tensor squeeze(const Tensor & self, int64_t dim) const override; | |
virtual Tensor & squeeze_(Tensor & self) const override; | |
virtual Tensor & squeeze_(Tensor & self, int64_t dim) const override; | |
virtual Tensor sspaddmm(const Tensor & self, const Tensor & mat1, const Tensor & mat2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor & sspaddmm_out(Tensor & result, const Tensor & self, const Tensor & mat1, const Tensor & mat2, Scalar beta, Scalar alpha) const override; | |
virtual Tensor stack(TensorList tensors, int64_t dim) const override; | |
virtual Tensor & stack_out(Tensor & result, TensorList tensors, int64_t dim) const override; | |
virtual Tensor stft(const Tensor & self, int64_t frame_length, int64_t hop, int64_t fft_size, bool normalized, bool onesided, const Tensor & window, int64_t pad_end) const override; | |
virtual int64_t stride(const Tensor & self, int64_t dim) const override; | |
virtual Tensor sum(const Tensor & self, ScalarType dtype) const override; | |
virtual Tensor sum(const Tensor & self) const override; | |
virtual Tensor _sum(const Tensor & self) const override; | |
virtual Tensor sum(const Tensor & self, IntList dim, bool keepdim, ScalarType dtype) const override; | |
virtual Tensor sum(const Tensor & self, IntList dim, bool keepdim) const override; | |
virtual Tensor sum(const Tensor & self, IntList dim, ScalarType dtype) const override; | |
virtual Tensor _sum(const Tensor & self, IntList dim, bool keepdim) const override; | |
virtual Tensor & sum_out(Tensor & result, const Tensor & self, IntList dim, bool keepdim, ScalarType dtype) const override; | |
virtual Tensor & sum_out(Tensor & result, const Tensor & self, IntList dim, bool keepdim) const override; | |
virtual Tensor & sum_out(Tensor & result, const Tensor & self, IntList dim, ScalarType dtype) const override; | |
virtual Tensor & _sum_out(Tensor & result, const Tensor & self, IntList dim, bool keepdim) const override; | |
virtual Tensor & _sum_cuda_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor sqrt(const Tensor & self) const override; | |
virtual Tensor & sqrt_(Tensor & self) const override; | |
virtual Tensor & sqrt_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor prod(const Tensor & self, ScalarType dtype) const override; | |
virtual Tensor prod(const Tensor & self) const override; | |
virtual Tensor _prod(const Tensor & self) const override; | |
virtual Tensor prod(const Tensor & self, int64_t dim, bool keepdim, ScalarType dtype) const override; | |
virtual Tensor prod(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor prod(const Tensor & self, int64_t dim, ScalarType dtype) const override; | |
virtual Tensor _prod(const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor & prod_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim, ScalarType dtype) const override; | |
virtual Tensor & prod_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor & prod_out(Tensor & result, const Tensor & self, int64_t dim, ScalarType dtype) const override; | |
virtual Tensor & _prod_out(Tensor & result, const Tensor & self, int64_t dim, bool keepdim) const override; | |
virtual Tensor t(const Tensor & self) const override; | |
virtual Tensor & t_(Tensor & self) const override; | |
virtual Tensor tan(const Tensor & self) const override; | |
virtual Tensor & tan_(Tensor & self) const override; | |
virtual Tensor & tan_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor tanh(const Tensor & self) const override; | |
virtual Tensor & tanh_(Tensor & self) const override; | |
virtual Tensor & tanh_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor transpose(const Tensor & self, int64_t dim0, int64_t dim1) const override; | |
virtual Tensor & transpose_(Tensor & self, int64_t dim0, int64_t dim1) const override; | |
virtual Tensor _trilinear(const Tensor & i1, const Tensor & i2, const Tensor & i3, IntList expand1, IntList expand2, IntList expand3, IntList sumdim, int64_t unroll_dim) const override; | |
virtual Tensor triplet_margin_loss(const Tensor & anchor, const Tensor & positive, const Tensor & negative, double margin, double p, double eps, bool swap, bool size_average, bool reduce) const override; | |
virtual Tensor trunc(const Tensor & self) const override; | |
virtual Tensor & trunc_(Tensor & self) const override; | |
virtual Tensor & trunc_out(Tensor & result, const Tensor & self) const override; | |
virtual Tensor type_as(const Tensor & self, const Tensor & other) const override; | |
virtual std::tuple<Tensor,Tensor> _unique(const Tensor & self, bool sorted, bool return_inverse) const override; | |
virtual Tensor _unsafe_view(const Tensor & self, IntList size) const override; | |
virtual Tensor unsqueeze(const Tensor & self, int64_t dim) const override; | |
virtual Tensor & unsqueeze_(Tensor & self, int64_t dim) const override; | |
virtual Tensor view_as(const Tensor & self, const Tensor & other) const override; | |
virtual Tensor where(const Tensor & condition, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor _s_where(const Tensor & condition, const Tensor & self, const Tensor & other) const override; | |
virtual Tensor zeros(IntList size) const override; | |
virtual Tensor & zeros_out(Tensor & result, IntList size) const override; | |
virtual Tensor zeros_like(const Tensor & self) const override; | |
virtual Tensor zeros_like(const Tensor & self, const Type & dtype) const override; | |
virtual Tensor _standard_gamma_grad(const Tensor & self, const Tensor & output) const override; | |
virtual Tensor _standard_gamma(const Tensor & self, Generator * generator) const override; | |
virtual Tensor poisson(const Tensor & self, Generator * generator) const override; | |
private: | |
// checks that t is actually a Variable | |
static Variable & checked_cast_variable(const Tensor & t, const char * name, int pos); | |
static at::Tensor & unpack(const Tensor & t, const char * name, int pos); | |
static at::SparseTensorRef unpack(SparseTensorRef t, const char * name, int pos); | |
static at::Tensor unpack_opt(const Tensor & t, const char * name, int pos); | |
static std::vector<at::Tensor> unpack(at::TensorList tl, const char *name, int pos); | |
at::Type* baseType; | |
std::string str; | |
}; | |
}} // namespace torch::autograd |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment