torch.rst.txt

torch

.. automodule:: torch

Tensors

.. autofunction:: is_tensor

.. autofunction:: is_storage

.. autofunction:: set_default_tensor_type

.. autofunction:: numel

.. autofunction:: set_printoptions

Creation Ops

.. autofunction:: eye

.. autofunction:: from_numpy

.. autofunction:: linspace

.. autofunction:: logspace

.. autofunction:: ones

.. autofunction:: rand

.. autofunction:: randn

.. autofunction:: randperm

.. autofunction:: arange

.. autofunction:: range

.. autofunction:: zeros

Indexing, Slicing, Joining, Mutating Ops

.. autofunction:: cat

.. autofunction:: chunk

.. autofunction:: gather

.. autofunction:: index_select

.. autofunction:: masked_select

.. autofunction:: nonzero

.. autofunction:: split

.. autofunction:: squeeze

.. autofunction:: stack

.. autofunction:: t

.. autofunction:: transpose

.. autofunction:: unbind

.. autofunction:: unsqueeze

Random sampling

.. autofunction:: manual_seed

.. autofunction:: initial_seed

.. autofunction:: get_rng_state

.. autofunction:: set_rng_state

.. autodata:: default_generator

.. autofunction:: bernoulli

.. autofunction:: multinomial

.. autofunction:: normal

Serialization

.. autofunction:: save

.. autofunction:: load

Parallelism

.. autofunction:: get_num_threads

.. autofunction:: set_num_threads

Math operations

Pointwise Ops

.. autofunction:: abs

.. autofunction:: acos

.. autofunction:: add

.. autofunction:: addcdiv

.. autofunction:: addcmul

.. autofunction:: asin

.. autofunction:: atan

.. autofunction:: atan2

.. autofunction:: ceil

.. autofunction:: clamp

.. autofunction:: cos

.. autofunction:: cosh

.. autofunction:: div

.. autofunction:: exp

.. autofunction:: floor

.. autofunction:: fmod

.. autofunction:: frac

.. autofunction:: lerp

.. autofunction:: log

.. autofunction:: log1p

.. autofunction:: mul

.. autofunction:: neg

.. autofunction:: pow

.. autofunction:: reciprocal

.. autofunction:: remainder

.. autofunction:: round

.. autofunction:: rsqrt

.. autofunction:: sigmoid

.. autofunction:: sign

.. autofunction:: sin

.. autofunction:: sinh

.. autofunction:: sqrt

.. autofunction:: tan

.. autofunction:: tanh

.. autofunction:: trunc

Reduction Ops

.. autofunction:: cumprod

.. autofunction:: cumsum

.. autofunction:: dist

.. autofunction:: mean

.. autofunction:: median

.. autofunction:: mode

.. autofunction:: norm

.. autofunction:: prod

.. autofunction:: std

.. autofunction:: sum

.. autofunction:: var

Comparison Ops

.. autofunction:: eq

.. autofunction:: equal

.. autofunction:: ge

.. autofunction:: gt

.. autofunction:: kthvalue

.. autofunction:: le

.. autofunction:: lt

.. autofunction:: max

.. autofunction:: min

.. autofunction:: ne

.. autofunction:: sort

.. autofunction:: topk

Other Operations

.. autofunction:: cross

.. autofunction:: diag

.. autofunction:: histc

.. autofunction:: renorm

.. autofunction:: trace

.. autofunction:: tril

.. autofunction:: triu

BLAS and LAPACK Operations

.. autofunction:: addbmm

.. autofunction:: addmm

.. autofunction:: addmv

.. autofunction:: addr

.. autofunction:: baddbmm

.. autofunction:: bmm

.. autofunction:: btrifact

.. autofunction:: btrisolve

.. autofunction:: dot

.. autofunction:: eig

.. autofunction:: gels

.. autofunction:: geqrf

.. autofunction:: ger

.. autofunction:: gesv

.. autofunction:: inverse

.. autofunction:: mm

.. autofunction:: mv

.. autofunction:: orgqr

.. autofunction:: ormqr

.. autofunction:: potrf

.. autofunction:: potri

.. autofunction:: potrs

.. autofunction:: pstrf

.. autofunction:: qr

.. autofunction:: svd

.. autofunction:: symeig

.. autofunction:: trtrs