torch

Tensors

Creation Ops

Indexing, Slicing, Joining, Mutating Ops

Random sampling

In-place random sampling

There are a few more in-place random sampling functions defined on Tensors as well. Click through to refer to their documentation:

  • torch.Tensor.bernoulli_() - in-place version of torch.bernoulli()
  • torch.Tensor.cauchy_() - numbers drawn from the Cauchy distribution
  • torch.Tensor.exponential_() - numbers drawn from the exponential distribution
  • torch.Tensor.geometric_() - elements drawn from the geometric distribution
  • torch.Tensor.log_normal_() - samples from the log-normal distribution
  • torch.Tensor.normal_() - in-place version of torch.normal()
  • torch.Tensor.random_() - numbers sampled from the discrete uniform distribution
  • torch.Tensor.uniform_() - numbers sampled from the uniform distribution

Serialization

Parallelism

Math operations

Pointwise Ops

Reduction Ops

Comparison Ops

Other Operations

BLAS and LAPACK Operations