pytorch学习手册【二】

 

九、Reduction Ops（规约/简化操作）

torch.argmax(input, dim=None, keepdim=False)

torch.argmin(input, dim=None, keepdim=False)

 

torch.cumprod(input, dim, dtype=None) → Tensor

torch.cumsum(input, dim, out=None, dtype=None) → Tensor

torch.dist(input, other, p=2) → Tensor

torch.logsumexp(input, dim, keepdim=False, out=None)

torch.mean(input, dim, keepdim=False, out=None) → Tensor

torch.median()

torch.median(input) → Tensor

torch.mode(input, dim=-1, keepdim=False, values=None, indices=None) -> (Tensor, LongTensor)

torch.norm(input, p='fro', dim=None, keepdim=False, out=None)

torch.prod(input, dim, keepdim=False, dtype=None) → Tensor

torch.std()

torch.std(input, unbiased=True) → Tensor

torch.std(input, dim, keepdim=False, unbiased=True, out=None) → Tensor

torch.sum()

torch.sum(input, dtype=None) → Tensor

torch.sum(input, dim, keepdim=False, dtype=None) → Tensor

torch.unique(input, sorted=False, return_inverse=False, dim=None)[SOURCE]

torch.var()

torch.var(input, unbiased=True) → Tensor

torch.var(input, dim, keepdim=False, unbiased=True, out=None) → Tensor

十、Comparison Ops（比较操作）

torch.allclose(self, other, rtol=1e-05, atol=1e-08, equal_nan=False) → bool

torch.argsort(input, dim=None, descending=False)

torch.eq(input, other, out=None) → Tensor

torch.equal(tensor1, tensor2) → bool

torch.ge(input, other, out=None) → Tensor

torch.gt(input, other, out=None) → Tensor

torch.isfinite(tensor)

torch.isinf(tensor)

torch.isnan(tensor)

torch.kthvalue(input, k, dim=None, keepdim=False, out=None) -> (Tensor, LongTensor)

torch.le(input, other, out=None) → Tensor

torch.lt(input, other, out=None) → Tensor

torch.max()

torch.max(input) → Tensor

torch.max(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)

torch.max(input, other, out=None) → Tensor

torch.min()

torch.min(input) → Tensor

torch.min(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)

torch.min(input, other, out=None) → Tensor

torch.ne(input, other, out=None) → Tensor

torch.sort(input, dim=None, descending=False, out=None) -> (Tensor, LongTensor)

torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)

十一、Spectral Ops（信号处理相关的谱运算）

torch.fft(input, signal_ndim, normalized=False) → Tensor

torch.ifft(input, signal_ndim, normalized=False) → Tensor

torch.rfft(input, signal_ndim, normalized=False, onesided=True) → Tensor

torch.irfft(input, signal_ndim, normalized=False, onesided=True, signal_sizes=None) → Tensor

torch.stft(input, n_fft, hop_length=None, win_length=None, window=None, center=True, pad_mode='reflect', normalized=False, onesided=True)

torch.bartlett_window(window_length, periodic=True, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

torch.blackman_window(window_length, periodic=True, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

torch.hamming_window(window_length, periodic=True, alpha=0.54, beta=0.46, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

torch.hann_window(window_length, periodic=True, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

十二、Other Operations（其他操作）

torch.bincount(self, weights=None, minlength=0) → Tensor

torch.broadcast_tensors(*tensors) → List of Tensors[SOURCE]

torch.cross(input, other, dim=-1, out=None) → Tensor

torch.diag(input, diagonal=0, out=None) → Tensor

torch.diagonal() always returns the diagonal of its input.

torch.diagflat() always constructs a tensor with diagonal elements specified by the input.

torch.diag_embed(input, offset=0, dim1=-2, dim2=-1) → Tensor

torch.diagflat(input, diagonal=0) → Tensor

torch.diagonal(input, offset=0, dim1=0, dim2=1) → Tensor

torch.einsum(equation, *operands) → Tensor[SOURCE]

torch.flatten(input, start_dim=0, end_dim=-1) → Tensor

torch.flip(input, dims) → Tensor

torch.histc(input, bins=100, min=0, max=0, out=None) → Tensor

torch.meshgrid(*tensors, **kwargs)[SOURCE]

torch.renorm(input, p, dim, maxnorm, out=None) → Tensor

torch.roll(input, shifts, dims=None) → Tensor

torch.tensordot(a, b, dims=2)[SOURCE]

torch.trace(input) → Tensor

torch.tril(input, diagonal=0, out=None) → Tensor

torch.triu(input, diagonal=0, out=None) → Tensor

十三、BLAS and LAPACK Operations（线性代数相关的运算）

BLAS即 basic linear algebra subprogram

LAPACK即 Linear Algebra PACKage

 

torch.addbmm(beta=1, mat, alpha=1, batch1, batch2, out=None) → Tensor

torch.addmm(beta=1, mat, alpha=1, mat1, mat2, out=None) → Tensor

torch.addmv(beta=1, tensor, alpha=1, mat, vec, out=None) → Tensor

torch.addr(beta=1, mat, alpha=1, vec1, vec2, out=None) → Tensor

torch.baddbmm(beta=1, mat, alpha=1, batch1, batch2, out=None) → Tensor

torch.bmm(batch1, batch2, out=None) → Tensor

 

torch.btrifact(A, info=None, pivot=True)

torch.btrifact_with_info(A, pivot=True) -> (Tensor, IntTensor, IntTensor)

torch.btrisolve(b, LU_data, LU_pivots) → Tensor

torch.btriunpack(LU_data, LU_pivots, unpack_data=True, unpack_pivots=True)

torch.chain_matmul(*matrices)

torch.cholesky(A, upper=False, out=None) → Tensor

torch.dot(tensor1, tensor2) → Tensor

torch.eig(a, eigenvectors=False, out=None) -> (Tensor, Tensor)

torch.gels(B, A, out=None) → Tensor

torch.geqrf(input, out=None) -> (Tensor, Tensor)

torch.ger(vec1, vec2, out=None) → Tensor

torch.gesv(B, A) -> (Tensor, Tensor)

torch.inverse(input, out=None) → Tensor

torch.det(A) → Tensor

torch.logdet(A) → Tensor

torch.slogdet(A) -> (Tensor, Tensor)

torch.matmul(tensor1, tensor2, out=None) → Tensor

torch.matrix_power(input, n) → Tensor

torch.matrix_rank(input, tol=None, bool symmetric=False) → Tensor

torch.mm(mat1, mat2, out=None) → Tensor

torch.mv(mat, vec, out=None) → Tensor

torch.orgqr(a, tau) → Tensor

torch.pinverse(input, rcond=1e-15) → Tensor

torch.potrf(a, upper=True, out=None)

torch.potrs(b, u, upper=True, out=None) → Tensor

torch.pstrf(a, upper=True, out=None) -> (Tensor, Tensor)

torch.qr(input, out=None) -> (Tensor, Tensor)

torch.svd(input, some=True, compute_uv=True, out=None) -> (Tensor, Tensor, Tensor)

torch.symeig(input, eigenvectors=False, upper=True, out=None) -> (Tensor, Tensor)