Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
torch.tensor.Tensor Class Reference
Inheritance diagram for torch.tensor.Tensor:

Public Member Functions

def __deepcopy__ (self, memo)
 
def __reduce_ex__ (self, proto)
 
def __setstate__ (self, state)
 
def __repr__ (self)
 
def backward (self, gradient=None, retain_graph=None, create_graph=False)
 
def register_hook (self, hook)
 
def reinforce (self, reward)
 
def retain_grad (self)
 
def is_pinned (self)
 
def is_shared (self)
 
def share_memory_ (self)
 
def __reversed__ (self)
 
def norm (self, p="fro", dim=None, keepdim=False, dtype=None)
 
def potrf (self, upper=True)
 
def pstrf (self, upper=True)
 
def potrs (self, u, upper=True)
 
def gesv (self, A)
 
def stft (self, n_fft, hop_length=None, win_length=None, window=None, center=True, pad_mode='reflect', normalized=False, onesided=True)
 
def resize (self, sizes)
 
def resize_as (self, tensor)
 
def split (self, split_size, dim=0)
 
def unique (self, sorted=True, return_inverse=False, dim=None)
 
def __rsub__ (self, other)
 
def __rdiv__ (self, other)
 
def __format__ (self, format_spec)
 
def __ipow__ (self, other)
 
def __rpow__ (self, other)
 
def __floordiv__ (self, other)
 
def __rfloordiv__ (self, other)
 
def __len__ (self)
 
def __iter__ (self)
 
def __hash__ (self)
 
def __dir__ (self)
 
def __array__ (self, dtype=None)
 
def __array_wrap__ (self, array)
 
def __contains__ (self, element)
 
def __cuda_array_interface__ (self)
 

Public Attributes

 data
 
 retains_grad
 

Static Public Attributes

 detach
 
 detach_
 

Detailed Description

Definition at line 20 of file tensor.py.

Member Function Documentation

def torch.tensor.Tensor.__cuda_array_interface__ (   self)
Array view description for cuda tensors.

See:
https://numba.pydata.org/numba-doc/latest/cuda/cuda_array_interface.html

Definition at line 446 of file tensor.py.


The documentation for this class was generated from the following file: