blob: 5df05c7a111c886c84a06482fc6ac02f4c050915 [file] [log] [blame]
[[
name: THPTensor_(elementSize)
python_name: element_size
cpu_half: True
auto_gpu: False
only_register: True
]]
[[
name: THPTensor_(storage)
python_name: storage
cpu_half: True
auto_gpu: False
only_register: True
]]
[[
name: storageOffset
python_name: storage_offset
cpu_half: True
auto_gpu: False
return: long
arguments:
- THTensor* self
]]
[[
name: nDimension
python_name: ndimension
cpu_half: True
auto_gpu: False
return: long
arguments:
- THTensor* self
]]
[[
name: THPTensor_(nDimension)
python_name: dim
cpu_half: True
auto_gpu: False
only_register: True
method_flags: METH_KEYWORDS
]]
[[
python_name: index
name: THPTensor_(getValue)<true>
only_register: True
override_method_flags: METH_O
]]
[[
python_name: _set_index
name: THPTensor_(setIndex)
only_register: True
]]
[[
python_name: _check_advanced_indexing
name: THPTensor_(checkAdvancedIndexing)
cpu_half: False
only_register: True
override_method_flags: METH_O
]]
[[
python_name: _advanced_index_add
name: THPTensor_(advancedIndexAdd)
cpu_half: False
only_register: True
]]
[[
python_name: _advanced_index_select
name: THPTensor_(advancedIndexSelect)
cpu_half: False
only_register: True
]]
[[
name: resize_
return: self
cname: resize
cpu_half: True
before_call:
THPUtils_assert(arg_self->storage->flag & TH_STORAGE_RESIZABLE,
"calling resize_ on a tensor that has non-resizable storage. Clone it first "
"or create a new tensor instead.");
arguments:
- THTensor* self
- arg: THSize* size
long_args: True
- CONSTANT NULL
]]
[[
name: zeros
variants:
- function
auto_gpu: False
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THSize* size
long_args: True
]]
[[
name: zeros_like
cname: zerosLike
variants:
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* input
]]
[[
name: ones
variants:
- function
auto_gpu: False
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THSize* size
long_args: True
]]
[[
name: ones_like
cname: onesLike
variants:
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* input
]]
[[
name: numel
return: long
cname: nElement
cpu_half: True
auto_gpu: False
variants:
- method
- function
arguments:
- THTensor* self
]]
[[
name: THPTensor_(numel)
python_name: nelement
cpu_half: True
auto_gpu: False
only_register: True
method_flags: METH_KEYWORDS
]]
[[
name: set_
cname: set
cpu_half: True
auto_gpu: False
return: argument 0
options:
- cname: set
arguments:
- THTensor* self
- THTensor* source
- cname: setStorage
arguments:
- THTensor* self
- CONSTANT NULL, 0, NULL, NULL
- cname: setStorage
before_call: THLongStoragePtr __storage_size(THLongStorage_newWithSize1(THStorage_(size)(LIBRARY_STATE arg_storage)));
arguments:
- THTensor* self
- THStorage* storage
- CONSTANT 0
- CONSTANT __storage_size.get()
- CONSTANT NULL
- cname: setStorage
arguments:
- THTensor* self
- THStorage* sourceStorage
- long storage_offset
- THSize* size
- arg: THStride* stride
default: NULL
]]
[[
name: THPTensor_(select)
python_name: select
cpu_half: True
auto_gpu: False
only_register: True
]]
[[
name: THPTensor_(size)
python_name: size
cpu_half: True
auto_gpu: False
method_flags: METH_KEYWORDS
only_register: True
]]
[[
name: THPTensor_(stride)
python_name: stride
cpu_half: True
auto_gpu: False
method_flags: METH_KEYWORDS
only_register: True
]]
[[
name: fill_
cname: fill
return: self
arguments:
- THTensor* self
- real value
]]
[[
name: isSameSizeAs
python_name: is_same_size
cpu_half: True
auto_gpu: False
return: bool
arguments:
- THTensor* self
- THTensor* other
]]
[[
name: isContiguous
python_name: is_contiguous
cpu_half: True
auto_gpu: False
return: bool
arguments:
- THTensor* self
]]
[[
name: isSetTo
python_name: is_set_to
cpu_half: True
auto_gpu: False
return: bool
arguments:
- THTensor* self
- THTensor* tensor
]]
[[
name: maskedFill_
cname: maskedFill
python_name: masked_fill_
return: self
arguments:
- arg: THTensor* self
broadcast: mask inplace fallback types:Byte
- THBoolTensor* mask
- real value
]]
[[
name: maskedCopy_
cname: maskedCopy
python_name: masked_scatter_
return: self
arguments:
- arg: THTensor* self
broadcast: mask inplace fallback types:Byte
- THBoolTensor* mask
- THTensor* source
]]
[[
name: maskedSelect
python_name: masked_select
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: mask fallback types:Byte
- THBoolTensor* mask
]]
[[
name: transpose
variants:
- method
- function
cname: newTranspose
cpu_half: True
auto_gpu: False
return: THTensor*
arguments:
- THTensor* self
- arg: long dim0
wrap_dim: self
- arg: long dim1
wrap_dim: self
]]
[[
name: transpose_
cname: transpose
cpu_half: True
auto_gpu: False
return: self
arguments:
- THTensor* self
- THTensor* self
- arg: long dim0
wrap_dim: self
- arg: long dim1
wrap_dim: self
]]
[[
name: t
variants:
- method
- function
auto_gpu: False
cname: newTranspose
return: THTensor*
before_call: |
long ndim = arg_self->nDimension;
THPUtils_assert(ndim == 2, "t() expects a 2D tensor, but self is %ldD", ndim);
arguments:
- THTensor* self
- CONSTANT 0
- CONSTANT 1
]]
[[
name: t_
cname: transpose
auto_gpu: False
return: self
before_call: |
long ndim = arg_self->nDimension;
THPUtils_assert(ndim == 2, "t_() expects a 2D tensor, but self is %ldD", ndim);
arguments:
- THTensor* self
- THTensor* self
- CONSTANT 0
- CONSTANT 1
]]
[[
name: squeeze
cpu_half: True
variants:
- method
- function
return: argument 0
options:
- arguments:
- arg: THTensor* result
output: True
- THTensor* self
- cname: squeeze1d
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
]]
[[
name: squeeze_
cpu_half: True
return: self
options:
- cname: squeeze
arguments:
- THTensor* self
- THTensor* self
- cname: squeeze1d
arguments:
- THTensor* self
- THTensor* self
- arg: long dim
wrap_dim: self
]]
[[
name: unsqueeze
variants:
- method
- function
cpu_half: True
auto_gpu: False
return: argument 0
cname: unsqueeze1d
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long dim
wrap_dim: self+1
]]
[[
name: unsqueeze_
cpu_half: True
auto_gpu: False
return: self
cname: unsqueeze1d
arguments:
- THTensor* self
- THTensor* self
- arg: long dim
wrap_dim: self+1
]]
[[
name: nonzero
variants:
- method
- function
return: argument 0
arguments:
- arg: THIndexTensor* result
output: True
- THTensor* self
]]
[[
name: contiguous
cname: newContiguous
return: THTensor*
arguments:
- THTensor* self
]]
[[
name: clone
cname: newClone
return: THTensor*
aten_sparse: True
arguments:
- THTensor* self
]]
[[
name: view
cname: newView
auto_gpu: False
return: THTensor*
arguments:
- THTensor* self
- arg: THSize* size
long_args: True
]]
[[
name: expand
cname: newExpand
return: THTensor*
arguments:
- THTensor* self
- arg: THSize* size
long_args: True
]]
[[
name: resizeAs_
python_name: resize_as_
cname: resizeAs
return: self
arguments:
- THTensor* self
- THTensor* the_template
]]
[[
name: indexSelect
python_name: index_select
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
- THIndexTensor* index
]]
[[
name: indexCopy_
python_name: index_copy_
cname: indexCopy
return: argument 0
arguments:
- THTensor* self
- arg: long dim
wrap_dim: self
- THIndexTensor* index
- THTensor* source
]]
[[
name: indexAdd_
python_name: index_add_
cname: indexAdd
return: argument 0
arguments:
- THTensor* self
- arg: long dim
wrap_dim: self
- THIndexTensor* index
- THTensor* source
]]
[[
name: indexFill_
python_name: index_fill_
cname: indexFill
return: argument 0
arguments:
- THTensor* self
- arg: long dim
wrap_dim: self
- THIndexTensor* index
- real value
]]
[[
name: narrow
cpu_half: True
auto_gpu: False
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long dimension
wrap_dim: self
- long start
- long length
]]
[[
name: unfold
cpu_half: True
auto_gpu: False
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long dimension
wrap_dim: self
- long size
- long step
]]
[[
name: range
variants:
- function
backends:
- CPU
- CUDA
return: argument 0
before_arg_assign: |
PyErr_WarnEx(PyExc_UserWarning, "torch.range is deprecated in favor of torch.arange "
"and will be removed in 0.3. Note that arange generates values in [start; end), "
"not [start; end].", 1);
arguments:
- arg: THTensor* result
output: True
- accreal start
- accreal end
- arg: accreal step
default: 1
]]
[[
name: arange
variants:
- function
backends:
- CPU
- CUDA
return: argument 0
arguments:
- arg: THTensor* result
output: True
- accreal start
- accreal end
- arg: accreal step
default: 1
]]
[[
name: scatter_
return: argument 0
options:
- cname: scatter
arguments:
- THTensor* self
- arg: long dim
wrap_dim: self
- THIndexTensor* index
- THTensor* src
- cname: scatterFill
arguments:
- THTensor* self
- arg: long dim
wrap_dim: self
- THIndexTensor* index
- real value
]]
[[
name: scatter_add_
return: argument 0
cname: scatterAdd
arguments:
- THTensor* self
- arg: long dim
wrap_dim: self
- THIndexTensor* index
- THTensor* src
]]
[[
name: gather
variants:
- method
- function
return: argument 0
before_call: |
THLongStoragePtr _size(THIndexTensor_(newSizeOf)(LIBRARY_STATE arg_index));
THTensor_(resize)(LIBRARY_STATE arg_result, _size, NULL);
arguments:
- arg: THTensor* result
output: True
resize: index
- THTensor* self
- arg: long dim
wrap_dim: self
- THIndexTensor* index
]]
[[
name: THPTensor_stateless_(cat)
python_name: cat
method_flags: METH_KEYWORDS
only_register: True
variants:
- function
]]
[[
name: data_ptr
defined_if: "!IS_DISTRIBUTED"
with_gil: True
auto_gpu: False
return: void*
cpu_half: True
cname: data
arguments:
- THTensor* self
]]
[[
name: equal
variants:
- method
- function
return: bool
arguments:
- THTensor* self
- THTensor* other
]]
[[
python_name: copy_
name: THPTensor_(copy_)
cpu_half: True
method_flags: METH_KEYWORDS
only_register: True
]]
[[
name: __and__
variants:
- method
- function
return: argument 0
options:
- cname: bitand
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- cname: cbitand
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: __iand__
variants:
- method
- function
return: argument 0
options:
- cname: bitand
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: cbitand
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: __or__
variants:
- method
- function
return: argument 0
options:
- cname: bitor
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- cname: cbitor
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: __ior__
variants:
- method
- function
return: argument 0
options:
- cname: bitor
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: cbitor
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: __xor__
variants:
- method
- function
return: argument 0
options:
- cname: bitxor
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- cname: cbitxor
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: __ixor__
variants:
- method
- function
return: argument 0
options:
- cname: bitxor
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: cbitxor
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: __lshift__
variants:
- method
- function
return: argument 0
options:
- cname: lshift
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- cname: clshift
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: __ilshift__
variants:
- method
- function
return: argument 0
options:
- cname: lshift
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: clshift
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: __rshift__
variants:
- method
- function
return: argument 0
options:
- cname: rshift
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- cname: crshift
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: __irshift__
variants:
- method
- function
return: argument 0
options:
- cname: rshift
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: crshift
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: THPTensor_(apply)
python_name: apply_
defined_if: "!IS_DISTRIBUTED"
backends:
- CPU
cpu_half: True
only_register: True
override_method_flags: METH_O
]]
[[
name: THPTensor_(map)
python_name: map_
defined_if: "!IS_DISTRIBUTED"
backends:
- CPU
cpu_half: True
only_register: True
]]
[[
name: THPTensor_(map2)
python_name: map2_
defined_if: "!IS_DISTRIBUTED"
backends:
- CPU
cpu_half: True
only_register: True
]]
[[
name: lt
variants:
- method
- function
return: argument 0
options:
- cname: ltValue
arguments:
- arg: THBoolTensor* result
output: True
- THTensor* self
- real value
- cname: ltTensor
arguments:
- arg: THBoolTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: lt_
return: self
options:
- cname: ltValueT
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: ltTensorT
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- arg: THTensor* other
]]
[[
name: gt
variants:
- method
- function
return: argument 0
options:
- cname: gtValue
arguments:
- arg: THBoolTensor* result
output: True
- THTensor* self
- real value
- cname: gtTensor
arguments:
- arg: THBoolTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: gt_
return: self
options:
- cname: gtValueT
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: gtTensorT
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: le
variants:
- method
- function
return: argument 0
options:
- cname: leValue
arguments:
- arg: THBoolTensor* result
output: True
- THTensor* self
- real value
- cname: leTensor
arguments:
- arg: THBoolTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: le_
return: self
options:
- cname: leValueT
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: leTensorT
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: ge
variants:
- method
- function
return: argument 0
options:
- cname: geValue
arguments:
- arg: THBoolTensor* result
output: True
- THTensor* self
- real value
- cname: geTensor
arguments:
- arg: THBoolTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: ge_
return: self
options:
- cname: geValueT
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: geTensorT
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: eq
variants:
- method
- function
return: argument 0
options:
- cname: eqValue
arguments:
- arg: THBoolTensor* result
output: True
- THTensor* self
- real value
- cname: eqTensor
arguments:
- arg: THBoolTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: eq_
return: self
options:
- cname: eqValueT
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: eqTensorT
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: ne
variants:
- method
- function
return: argument 0
options:
- cname: neValue
arguments:
- arg: THBoolTensor* result
output: True
- THTensor* self
- real value
- cname: neTensor
arguments:
- arg: THBoolTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: ne_
return: self
options:
- cname: neValueT
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: neTensorT
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: min
variants:
- method
- function
options:
- cname: minall
return: real
arguments:
- THTensor* self
- cname: cmin
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
- cname: min
return: argument 0,1
arguments:
- arg: THTensor* min
output: True
- arg: THIndexTensor* min_indices
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
- arg: bool keepdim
default: "false"
]]
[[
name: max
variants:
- method
- function
options:
- cname: maxall
return: real
arguments:
- THTensor* self
- cname: cmax
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
- cname: max
return: argument 0,1
arguments:
- arg: THTensor* max
output: True
- arg: THIndexTensor* max_indices
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
- arg: bool keepdim
default: "false"
]]
[[
name: kthvalue
backends:
- CPU
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* values
output: True
- arg: THIndexTensor* indices
output: True
- THTensor* self
- long k
- arg: long dim
wrap_dim: self
default: __last_dim
- arg: bool keepdim
default: "false"
]]
[[
name: mode
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* values
output: True
- arg: THIndexTensor* indices
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
default: __last_dim
- arg: bool keepdim
default: "false"
]]
[[
name: median
variants:
- method
- function
return: argument 0,1
options:
- cname: medianall
return: real
arguments:
- THTensor* self
- cname: median
arguments:
- arg: THTensor* values
output: True
- arg: THIndexTensor* indices
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
- arg: bool keepdim
default: "false"
]]
[[
name: sort
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* values
output: True
- arg: THIndexTensor* indices
output: True
- THTensor* self
- arg: long dim
default: __last_dim
wrap_dim: self
- arg: bool descending
default: "false"
]]
[[
name: topk
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* values
output: True
- arg: THIndexTensor* indices
output: True
- THTensor* self
- long k
- arg: long dim
default: __last_dim
wrap_dim: self
- arg: bool largest
default: "true"
- arg: bool sorted
default: "true"
]]
[[
name: all
types:
- Byte
backends:
- CPU
- CUDA
cname: logicalall
return: bool
arguments:
- THTensor* self
]]
[[
name: any
types:
- Byte
backends:
- CPU
- CUDA
cname: logicalany
return: bool
arguments:
- THTensor* self
]]
[[
name: getDevice
python_name: get_device
backends:
- CUDA
return: long
arguments:
- THTensor* self
]]
[[
name: THPTensor_(new)
python_name: new
method_flags: METH_KEYWORDS
backends:
- CUDA
only_register: True
]]
[[
name: THPTensor_(recordStream)
python_name: record_stream
override_method_flags: METH_O
backends:
- CUDA
only_register: True
]]
[[
name: abs
return: argument 0
types:
- floating_point
- Long
- Int
- Short
backends:
- CPU
- CUDA
variants:
- method
- function
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
]]
[[
name: abs_
cname: abs
return: self
types:
- floating_point
- Long
- Int
- Short
backends:
- CPU
- CUDA
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: sigmoid_
types:
- floating_point
backends:
- CPU
- CUDA
cname: sigmoid
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: sigmoid
types:
- floating_point
backends:
- CPU
- CUDA
cname: sigmoid
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: log_
types:
- floating_point
backends:
- CPU
- CUDA
cname: log
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: log
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: log1p_
types:
- floating_point
backends:
- CPU
- CUDA
cname: log1p
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: log1p
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: lgamma
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: lgamma_
types:
- floating_point
backends:
- CPU
- CUDA
cname: lgamma
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: exp_
types:
- floating_point
backends:
- CPU
- CUDA
cname: exp
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: exp
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: cos_
types:
- floating_point
backends:
- CPU
- CUDA
cname: cos
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: cos
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: acos_
types:
- floating_point
backends:
- CPU
- CUDA
cname: acos
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: acos
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: cosh_
types:
- floating_point
backends:
- CPU
- CUDA
cname: cosh
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: cosh
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: sin_
types:
- floating_point
backends:
- CPU
- CUDA
cname: sin
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: sin
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: asin_
types:
- floating_point
backends:
- CPU
- CUDA
cname: asin
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: asin
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: sinh_
types:
- floating_point
backends:
- CPU
- CUDA
cname: sinh
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: sinh
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: tan_
types:
- floating_point
backends:
- CPU
- CUDA
cname: tan
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: tan
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: atan_
types:
- floating_point
backends:
- CPU
- CUDA
cname: atan
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: atan
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: tanh_
types:
- floating_point
backends:
- CPU
- CUDA
cname: tanh
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: tanh
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: sqrt_
types:
- floating_point
backends:
- CPU
- CUDA
cname: sqrt
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: sqrt
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: rsqrt_
types:
- floating_point
backends:
- CPU
- CUDA
cname: rsqrt
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: rsqrt
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: ceil_
types:
- floating_point
backends:
- CPU
- CUDA
cname: ceil
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: ceil
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: floor_
types:
- floating_point
backends:
- CPU
- CUDA
cname: floor
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: floor
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: round_
types:
- floating_point
backends:
- CPU
- CUDA
cname: round
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: round
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: trunc_
types:
- floating_point
backends:
- CPU
- CUDA
cname: trunc
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: trunc
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: frac_
types:
- floating_point
backends:
- CPU
- CUDA
cname: frac
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: frac
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: mean
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
options:
- cname: meanall
return: accreal
arguments:
- THTensor* self
- cname: mean
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
- arg: bool keepdim
default: "false"
]]
[[
name: var
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
options:
- cname: varall
return: accreal
arguments:
- THTensor* self
- arg: bool unbiased
if_true: 0
if_false: 1
default: 0
- cname: var
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
- arg: bool unbiased
if_true: 0
if_false: 1
default: 0
- arg: bool keepdim
default: "false"
]]
[[
name: std
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
options:
- cname: stdall
return: accreal
arguments:
- THTensor* self
- arg: bool unbiased
if_true: 0
if_false: 1
default: 0
- cname: std
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
- arg: bool unbiased
if_true: 0
if_false: 1
default: 0
- arg: bool keepdim
default: "false"
]]
[[
name: norm
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
options:
- cname: normall
return: accreal
arguments:
- THTensor* self
- arg: real p
default: AS_REAL(2)
- cname: norm
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- real p
- arg: long dim
wrap_dim: self
- arg: bool keepdim
default: "false"
]]
[[
name: renorm
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- real p
- arg: long dim
wrap_dim: self
- real maxnorm
]]
[[
name: renorm_
types:
- floating_point
backends:
- CPU
- CUDA
cname: renorm
return: self
arguments:
- THTensor* self
- THTensor* self
- real p
- arg: long dim
wrap_dim: self
- real maxnorm
]]
[[
name: dist
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
options:
- cname: dist
return: accreal
arguments:
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
- arg: real p
default: AS_REAL(2)
]]
[[
name: reciprocal
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
options:
- cname: cinv
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
]]
[[
name: reciprocal_
types:
- floating_point
backends:
- CPU
- CUDA
options:
- cname: cinv
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: neg
types:
- floating_point
- Long
- Int
- Short
backends:
- CPU
- CUDA
variants:
- method
- function
options:
- cname: neg
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
]]
[[
name: neg_
types:
- floating_point
- Long
- Int
- Short
backends:
- CPU
- CUDA
options:
- cname: neg
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: atan2
types:
- Float
- Double
backends:
- CPU
backend_type_pairs: [[CPU,Float],[CPU,Double],[CUDA,Float]]
variants:
- method
- function
cname: atan2
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: atan2_
backend_type_pairs: [[CPU,Float],[CPU,Double],[CUDA,Float]]
cname: atan2
return: argument 0
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other fallback inplace
- THTensor* other
]]
[[
name: pow
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
options:
- cname: pow
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- real exponent
- cname: cpow
arguments:
- arg: THTensor* destination
output: True
- arg: THTensor* self
broadcast: exponent fallback
- THTensor* exponent
- cname: tpow
arguments:
- arg: THTensor* destination
output: True
- real base
- THTensor* self
]]
[[
name: pow_
types:
- floating_point
backends:
- CPU
- CUDA
return: argument 0
cname: pow
options:
- cname: pow
arguments:
- THTensor* self
- THTensor* self
- real exponent
- cname: cpow
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: exponent inplace fallback
- THTensor* exponent
]]
[[
name: lerp
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
cname: lerp
arguments:
- arg: THTensor* destination
output: True
- arg: THTensor* self
broadcast: end fallback
- THTensor* end
- real weight
]]
[[
name: lerp_
types:
- floating_point
backends:
- CPU
- CUDA
return: self
cname: lerp
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: end fallback inplace
- THTensor* end
- real weight
]]
[[
name: linspace
types:
- Float
- Double
backends:
- CPU
variants:
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- real start
- real end
- arg: long steps
default: 100
]]
[[
name: logspace
types:
- Float
- Double
backends:
- CPU
variants:
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- real start
- real end
- arg: long steps
default: 100
]]
[[
name: histc
types:
- Float
- Double
backends:
- CPU
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- arg: long bins
default: 100
- arg: real min
default: 0
- arg: real max
default: 0
]]
[[
name: zero_
cname: zero
return: self
arguments:
- THTensor* self
]]
[[
name: sum
variants:
- method
- function
options:
- cname: sumall
return: accreal
arguments:
- THTensor* self
- cname: sum
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
- arg: bool keepdim
default: "false"
]]
[[
name: prod
variants:
- method
- function
options:
- cname: prodall
return: accreal
arguments:
- THTensor* self
- cname: prod
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
- arg: bool keepdim
default: "false"
]]
[[
name: cumsum
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
]]
[[
name: cumprod
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long dim
wrap_dim: self
]]
[[
name: sign
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
]]
[[
name: sign_
cname: sign
return: self
arguments:
- THTensor* self
- THTensor* self
]]
[[
name: trace
variants:
- method
- function
return: accreal
arguments:
- THTensor* self
]]
[[
name: add
variants:
- method
- function
return: argument 0
options:
- cname: add_scaled
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- cname: cadd
aten_sparse: True
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* other
- sparse: True
cname: spcadd
aten_dense_sparse: True
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THSTensor* other
]]
[[
name: add_
return: argument 0
options:
- cname: add_scaled
arguments:
- THTensor* self
- THTensor* self
- real value
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- cname: cadd
aten_sparse: True
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* other
- sparse: True
cname: spcadd
aten_dense_sparse: True
arguments:
- THTensor* self
- THTensor* self
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THSTensor* other
]]
[[
name: sub
variants:
- method
- function
return: argument 0
options:
- cname: sub_scaled
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- cname: csub
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* other
]]
[[
name: sub_
return: argument 0
options:
- cname: sub_scaled
arguments:
- THTensor* self
- THTensor* self
- real value
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- cname: csub
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* other
]]
[[
name: mul
variants:
- method
- function
return: argument 0
options:
- cname: mul
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- cname: cmul
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- arg: THTensor* other
]]
[[
name: mul_
return: argument 0
options:
- cname: mul
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: cmul
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: div
variants:
- method
- function
return: argument 0
options:
- cname: div
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- cname: cdiv
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: div_
return: argument 0
options:
- cname: div
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: cdiv
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: fmod
return: argument 0
variants:
- method
- function
options:
- cname: fmod
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- cname: cfmod
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- THTensor* other
]]
[[
name: fmod_
return: argument 0
options:
- cname: fmod
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: cfmod
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: remainder
return: argument 0
variants:
- method
- function
options:
- cname: remainder
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- real value
- cname: cremainder
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: other fallback
- arg: THTensor* other
]]
[[
name: remainder_
return: argument 0
options:
- cname: remainder
arguments:
- THTensor* self
- THTensor* self
- real value
- cname: cremainder
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: other inplace fallback
- THTensor* other
]]
[[
name: clamp
variants:
- method
- function
return: argument 0
options:
- cname: clamp
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- real min
- real max
- cname: cmaxValue
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: real min
kwarg_only: True
- cname: cminValue
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: real max
kwarg_only: True
]]
[[
name: clamp_
cname: clamp
return: self
options:
- cname: clamp
arguments:
- THTensor* self
- THTensor* self
- real min
- real max
- cname: cmaxValue
arguments:
- THTensor* self
- THTensor* self
- arg: real min
kwarg_only: True
- cname: cminValue
arguments:
- THTensor* self
- THTensor* self
- arg: real max
kwarg_only: True
]]
[[
name: dot
backend_type_pairs: [[CUDA,floating_point], [CPU,all]]
variants:
- method
- function
return: accreal
arguments:
- arg: THTensor* self
assert_ndim: 1
- arg: THTensor* tensor
assert_ndim: 1
]]
[[
name: tril
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- arg: long diagonal
default: 0
]]
[[
name: tril_
cname: tril
return: self
arguments:
- THTensor* self
- THTensor* self
- arg: long diagonal
default: 0
]]
[[
name: triu
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- arg: long diagonal
default: 0
]]
[[
name: triu_
cname: triu
return: self
arguments:
- THTensor* self
- THTensor* self
- arg: long diagonal
default: 0
]]
[[
name: cross
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* destination
output: True
- THTensor* self
- THTensor* other
- arg: long dim
default: -1
]]
[[
name: eye
backends:
- CPU
- CUDA
variants:
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- long n
- arg: long m
default: 1
]]
[[
name: diag
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- arg: long diagonal
default: 0
]]
[[
name: addmm
variants:
- method
- function
return: argument 0
options:
- arguments:
- arg: THTensor* result
output: True
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- arg: THTensor* self
broadcast: mat1,mat2 dims:mat1.dim0,mat2.dim1
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* mat1
- THTensor* mat2
- cname: spaddmm
sparse: yes
arguments:
- arg: THTensor* result
output: True
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- THTensor* self
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THSTensor* mat1
- THTensor* mat2
]]
[[
name: addmm_
return: self
options:
- cname: addmm
arguments:
- THTensor* self
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- THTensor* self
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* mat1
- THTensor* mat2
- cname: spaddmm
sparse: yes
arguments:
- arg: THTensor* self
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- THTensor* self
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THSTensor* mat1
- THTensor* mat2
]]
[[
name: addmv
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- arg: THTensor* self
broadcast: mat,vec dims:mat.dim0
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* mat
- THTensor* vec
]]
[[
name: addmv_
cname: addmv
return: self
arguments:
- THTensor* self
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- THTensor* self
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* mat
- THTensor* vec
]]
[[
name: addr
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- arg: THTensor* self
broadcast: vec1,vec2 dims:vec1.dim0,vec2.dim0
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* vec1
- THTensor* vec2
]]
[[
name: addr_
cname: addr
return: self
arguments:
- THTensor* self
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- THTensor* self
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* vec1
- THTensor* vec2
]]
[[
name: ger
cname: addr
variants:
- method
- function
return: argument 0
before_call: |
long s1 = THTensor_(size)(LIBRARY_STATE ((THPTensor*)$arg4)->cdata, 0);
long s2 = THTensor_(size)(LIBRARY_STATE ((THPTensor*)$arg5)->cdata, 0);
THTensor_(resize2d)(LIBRARY_STATE ((THPTensor*)$arg0)->cdata, s1, s2);
arguments:
- arg: THTensor* result
output: True
resize: [ [self,0], [vec2,0] ]
- CONSTANT AS_REAL(0)
- argument 0
- CONSTANT AS_REAL(1)
- THTensor* self
- THTensor* vec2
]]
[[
name: mv
cname: addmv
variants:
- method
- function
return: argument 0
before_call: |
long s = THTensor_(size)(LIBRARY_STATE ((THPTensor*)$arg4)->cdata, 0);
THTensor_(resize1d)(LIBRARY_STATE ((THPTensor*)$arg0)->cdata, s);
#if !IS_CUDA
THTensor_(zero)(LIBRARY_STATE ((THPTensor*)$arg0)->cdata);
#endif
arguments:
- arg: THTensor* result
output: True
resize: [ [self, 0] ]
cpu_zero: True
- CONSTANT AS_REAL(0)
- argument 0
- CONSTANT AS_REAL(1)
- THTensor* self
- THTensor* vec
]]
[[
name: mm
variants:
- method
- function
return: argument 0
options:
- cname: addmm
before_call: |
long s1 = THTensor_(size)(LIBRARY_STATE ((THPTensor*)$arg4)->cdata, 0);
long s2 = THTensor_(size)(LIBRARY_STATE ((THPTensor*)$arg5)->cdata, 1);
THTensor_(resize2d)(LIBRARY_STATE ((THPTensor*)$arg0)->cdata, s1, s2);
#if !IS_CUDA
THTensor_(zero)(LIBRARY_STATE ((THPTensor*)$arg0)->cdata);
#endif
arguments:
- arg: THTensor* result
output: True
resize: [ [self, 0], [mat2,1] ]
cpu_zero: True
- CONSTANT AS_REAL(0)
- argument 0
- CONSTANT AS_REAL(1)
- THTensor* self
- THTensor* mat2
- cname: spaddmm
sparse: True
arguments:
- arg: THTensor* result
output: True
- CONSTANT AS_REAL(0)
- argument 0
- CONSTANT AS_REAL(1)
- THSTensor* self
- THTensor* mat2
]]
[[
name: bmm
cname: baddbmm
variants:
- method
- function
return: argument 0
before_call: |
long s1 = THTensor_(size)(LIBRARY_STATE ((THPTensor*)$arg4)->cdata, 0);
long s2 = THTensor_(size)(LIBRARY_STATE ((THPTensor*)$arg4)->cdata, 1);
long s3 = THTensor_(size)(LIBRARY_STATE ((THPTensor*)$arg5)->cdata, 2);
THTensor_(resize3d)(LIBRARY_STATE ((THPTensor*)$arg0)->cdata, s1, s2, s3);
#if !IS_CUDA
THTensor_(zero)(LIBRARY_STATE ((THPTensor*)$arg0)->cdata);
#endif
arguments:
- arg: THTensor* result
output: True
resize: [ [self,0], [self,1], [mat2,2] ]
cpu_zero: True
- CONSTANT AS_REAL(0)
- argument 0
- CONSTANT AS_REAL(1)
- THTensor* self
- THTensor* mat2
]]
[[
name: addbmm
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- arg: THTensor* self
broadcast: batch1,batch2 dims:batch1.dim1,batch2.dim2
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* batch1
- THTensor* batch2
]]
[[
name: addbmm_
cname: addbmm
return: self
arguments:
- THTensor* self
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- THTensor* self
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* batch1
- THTensor* batch2
]]
[[
name: baddbmm
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- arg: THTensor* self
broadcast: batch1,batch2 dims:batch1.dim0,batch1.dim1,batch2.dim2
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* batch1
- THTensor* batch2
]]
[[
name: baddbmm_
cname: baddbmm
return: argument 0
arguments:
- THTensor* self
- arg: real beta
default: AS_REAL(1)
kwarg_only: True
- THTensor* self
- arg: real alpha
default: AS_REAL(1)
kwarg_only: True
- THTensor* batch1
- THTensor* batch2
]]
[[
name: addcmul
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: tensor1,tensor2 fallback
- arg: real value
default: AS_REAL(1)
kwarg_only: True
- THTensor* tensor1
- THTensor* tensor2
]]
[[
name: addcmul_
options:
- cname: addcmul
return: argument 0
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: tensor1,tensor2 inplace fallback
- arg: real value
default: AS_REAL(1)
kwarg_only: True
- THTensor* tensor1
- THTensor* tensor2
- cname: spaddcmul
defined_if: "!IS_DISTRIBUTED"
return: argument 0
arguments:
- THTensor* self
- THTensor* self
- arg: real value
default: AS_REAL(1)
kwarg_only: True
- THSTensor* tensor1
- THSTensor* tensor2
]]
[[
name: addcdiv
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THTensor* self
broadcast: tensor1,tensor2 fallback
- arg: real value
default: AS_REAL(1)
kwarg_only: True
- THTensor* tensor1
- THTensor* tensor2
]]
[[
name: addcdiv_
cname: addcdiv
return: argument 0
arguments:
- THTensor* self
- arg: THTensor* self
broadcast: tensor1,tensor2 inplace fallback
- arg: real value
default: AS_REAL(1)
kwarg_only: True
- THTensor* tensor1
- THTensor* tensor2
]]
[[
name: gesv
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* solution
output: True
- arg: THTensor* lu
output: True
- THTensor* self
- THTensor* A
]]
[[
name: gels
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* res1
output: True
- arg: THTensor* res2
output: True
- THTensor* self
- THTensor* A
]]
[[
name: trtrs
types:
- Float
- Double
backends:
- CPU
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* res1
output: True
- arg: THTensor* res2
output: True
- THTensor* self
- THTensor* A
- arg: bool upper
if_true: U
if_false: L
default: U
- arg: bool transpose
if_true: T
if_false: N
default: N
- arg: bool unitriangular
if_true: U
if_false: N
default: N
]]
[[
name: symeig
cname: syev
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* res1
output: True
- arg: THTensor* res2
output: True
- THTensor* self
- arg: bool eigenvectors
if_true: V
if_false: N
default: N
- arg: bool upper
if_true: U
if_false: L
default: U
]]
[[
name: eig
cname: geev
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* res1
output: True
- arg: THTensor* res2
output: True
- THTensor* self
- arg: bool eigenvectors
if_true: V
if_false: N
default: N
]]
[[
name: svd
cname: gesvd
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0,1,2
arguments:
- arg: THTensor* res1
output: True
- arg: THTensor* res2
output: True
- arg: THTensor* res3
output: True
- THTensor* self
- arg: bool some
if_true: S
if_false: A
default: S
]]
[[
name: inverse
cname: getri
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* output
output: True
- THTensor* self
]]
[[
name: potrf
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* output
output: True
- THTensor* self
- arg: bool upper
if_true: U
if_false: L
default: U
]]
[[
name: potrs
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- THTensor* input2
- arg: bool upper
if_true: U
if_false: L
default: U
]]
[[
name: potri
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* output
output: True
- THTensor* self
- arg: bool upper
if_true: U
if_false: L
default: U
]]
[[
name: pstrf
types:
- Float
- Double
backends:
- CPU
variants:
- method
- function
return: argument 0,1
after_call:
THIntTensor_sub(((THPIntTensor*)$arg1)->cdata, ((THPIntTensor*)$arg1)->cdata, 1);
arguments:
- arg: THTensor* res1
output: True
- arg: THIntegerTensor* res2
output: True
- THTensor* self
- arg: bool upper
if_true: U
if_false: L
default: U
- arg: real tol
default: -1
]]
[[
name: qr
types:
- Float
- Double
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* res1
output: True
- arg: THTensor* res2
output: True
- THTensor* self
]]
[[
name: geqrf
types:
- Float
- Double
backends:
- CPU
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* res1
output: True
- arg: THTensor* res2
output: True
- THTensor* self
]]
[[
name: orgqr
types:
- Float
- Double
backends:
- CPU
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- THTensor* input2
]]
[[
name: ormqr
types:
- Float
- Double
backends:
- CPU
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- THTensor* input2
- THTensor* input3
- arg: bool left
if_true: L
if_false: R
default: L
- arg: bool transpose
if_true: T
if_false: N
default: N
]]
[[
name: btrifact
cname: btrifact
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0,1
arguments:
- arg: THTensor* result
output: True
- arg: THIntegerTensor* pivots
output: True
- arg: THIntegerTensor* info
kwarg_only: True
default: NULL
- arg: bool pivot
kwarg_only: True
default: "true"
- THTensor* self
]]
[[
name: btrisolve
cname: btrisolve
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- THTensor* self
- THTensor* LU_data
- THIntegerTensor* LU_pivots
]]
[[
name: randperm
defined_if: "!IS_DISTRIBUTED"
backends:
- CPU
variants:
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- long n
]]
[[
name: random_
defined_if: "!IS_DISTRIBUTED"
backends:
- CPU
return: self
options:
- cname: random
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- cname: cappedRandom
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- long to
- cname: clampedRandom
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- long from
- long to
]]
[[
name: multinomial
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- method
- function
return: argument 0
arguments:
- arg: THIndexTensor* result
output: True
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- THTensor* self
- long num_samples
- arg: bool replacement
default: "false"
]]
[[
name: uniform_
types:
- floating_point
backends:
- CPU
- CUDA
cname: uniform
return: self
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- arg: double from
default: 0
- arg: double to
default: 1
]]
[[
name: normal
types:
- floating_point
backends:
- CPU
- CUDA
return: argument 0
variants:
- function
options:
- cname: normal_means
arguments:
- arg: THTensor* output
output: True
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- THTensor* means
- arg: double std
default: 1
- cname: normal_stddevs
arguments:
- arg: THTensor* output
output: True
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- arg: double mean
- THTensor* std
- cname: normal_means_stddevs
arguments:
- arg: THTensor* output
output: True
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- THTensor* means
- THTensor* std
]]
[[
name: normal_
types:
- floating_point
backends:
- CPU
- CUDA
cname: normal
return: self
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- arg: double mean
default: 0
- arg: double std
default: 1
]]
[[
name: cauchy_
types:
- floating_point
backends:
- CPU
- CUDA
cname: cauchy
return: self
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- arg: double median
default: 0
- arg: double sigma
default: 1
]]
[[
name: logNormal_
cname: logNormal
python_name: log_normal_
types:
- floating_point
backends:
- CPU
- CUDA
return: self
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- arg: double mean
default: 1
- arg: double std
default: 2
]]
[[
name: exponential_
types:
- floating_point
backends:
- CPU
- CUDA
cname: exponential
return: self
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- arg: double lambd
default: 1
]]
[[
name: rand
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- arg: THSize* size
long_args: True
]]
[[
name: randn
types:
- floating_point
backends:
- CPU
- CUDA
variants:
- function
return: argument 0
arguments:
- arg: THTensor* result
output: True
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- arg: THSize* size
long_args: True
]]
[[
name: geometric_
defined_if: "!IS_DISTRIBUTED"
backends:
- CPU
- CUDA
cname: geometric
return: self
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- double p
]]
[[
name: bernoulli
types:
- Float
- Double
backends:
- CPU
return: argument 0
variants:
- method
- function
before_call:
THTensor_(resizeAs)(LIBRARY_STATE ((THPTensor*)$arg0)->cdata, ((THPTensor*)$arg2)->cdata);
cname: BERNOULLI_TENSOR
arguments:
- arg: THTensor* output
output: True
resize: self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- THTensor* self
]]
[[
name: bernoulli_
defined_if: "!IS_DISTRIBUTED"
backends:
- CPU
- CUDA
return: self
options:
- cname: bernoulli
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- arg: double p
default: 0.5
- cname: bernoulli_FloatTensor
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- BackendFloatTensor* float_p
- cname: bernoulli_DoubleTensor
arguments:
- THTensor* self
- arg: THGenerator* generator
default: THPDefaultGenerator->cdata
kwarg_only: True
- BackendDoubleTensor* float_p
]]