PyTorch-1.10(五)--torch数学运算

目录

元素操作

归纳操作

比较操作

频谱操作

其他操作

BLAS和LAPACK操作

---

元素操作

abs	计算输入中每个元素的绝对值。
absolute	torch.abs()别名
acos	计算输入中每个元素的反余弦。
arccos	torch.acos()别名
acosh	计算输入元素的反双曲余弦。
arccosh	torch.acosh()别名
add	将其他按alpha缩放的对象加到输入中. out=input+alpha×other，torch.add(a,c,alpah=2)
addcdiv	执行tensor1与tensor2的元素除法，将结果乘以标量值并将其加到输入中。
addcmul	执行tensor1与tensor2的元素相乘，将结果与标量值相乘，并将其加到输入中。
angle	计算给定输入张量的元素角度（弧度）。
asin	计算输入元素的反正弦。
arcsin	torch.asin()别名
asinh	计算输入元素的反双曲正弦。
arcsinh	torch.asinh()别名
atan	计算输入元素的反正切。
arctan	torch.atan()别名
atanh	计算输入元素的反双曲正切。
arctanh	torch.atanh()别名
atan2	input/other后，再计算反正切。
bitwise_not	计算给定输入张量的按位NOT。
bitwise_and	计算输入和其他数据的按位AND。
bitwise_or	计算输入和其他数据的按位OR。
bitwise_xor	计算输入和其他数据的按位异或XOR。
bitwise_left_shift	通过其他位计算输入的左算术移位。
bitwise_right_shift	通过其他位计算输入的右算术移位。
ceil	返回一个新的张量，其中输入中的元素为大于或等于每个元素的最小整数。
clamp	将输入中的所有元素clamp在范围内 [ min, max ].
clip	torch.clamp()别名
conj_physical	计算给定输入张量的元素共轭。
copysign	创建一个新的浮点张量，大小与输入元素一致，符号与其他元素的一致。
cos	返回带有输入元素余弦的新张量。
cosh	返回具有输入元素双曲余弦的新张量。
deg2rad	返回一个新的张量，其中输入的每个元素都从角度转换为弧度。
div	将输入的每个元素除以另一个元素的对应元素。
divide	torch.div()别名
digamma	torch.special.digamma()别名
erf	torch.special.erf()别名
erfc	torch.special.erfc()别名
erfinv	torch.special.erfinv()别名
exp	返回一个新的张量，其中包含输入张量输入元素的指数。
exp2	torch.special.exp2()别名
expm1	torch.special.expm1()别名
fake_quantize_per_channel_affine	返回一个新的张量，输入中的数据通过轴指定的通道，使用scale、zero_point、quant_min和quant_max对每个通道进行量化。
fake_quantize_per_tensor_affine	返回一个新的张量，其中输入的数据使用scale、zero point、quant min和quant max进行量化。
fix	torch.trunc()别名
float_power	以双精度将输入进行元素的指数幂运算。
floor	返回一个新的张量，该张量中元素小于或等于输入元素的最大整数。
floor_divide
fmod	​ 对浮点张量应用C++的std::fmod，对整型张量应用模运算。
frac	计算输入中每个元素的小数部分。
frexp	将输入分解为尾数和指数张量 input=mantissa×2exponent.
gradient	​ 使用二阶精确中心差分法估计函数g在一个或多个维度上的梯度。
imag	返回一个包含自张量虚值的新张量。
ldexp	将输入乘以 2**:attr:other.
lerp	两个张量的线性插值是否基于标量或张量权重开始（由输入给定）和结束，并返回结果输出张量。
lgamma	计算输入上gamma函数绝对值的自然对数。
log	返回具有输入元素自然对数的新张量。
log10	返回一个新的张量，其对数以输入元素的10为底。
log1p	返回一个自然对数为（1+input）的新张量。
log2	返回一个新的张量，其对数以输入元素的2为底。
logaddexp	输入的幂和的对数。
logaddexp2	以2为底的输入的幂和的对数。
logical_and	计算给定输入张量的元素逻辑AND。
logical_not	计算给定输入张量的元素逻辑NOT。
logical_or	计算给定输入张量的元素逻辑OR。
logical_xor	计算给定输入张量的元素逻辑异或。
logit	torch.special.logit()别名
hypot	给定直角三角形的边，返回斜边。
i0	torch.special.i0()别名
igamma	torch.special.gammainc()别名
igammac	torch.special.gammaincc()别名
mul	将输入乘以其他。
multiply	torch.mul()别名
mvlgamma	torch.special.multigammaln()别名
nan_to_num	将输入中的NaN、正无穷大和负无穷大值分别替换为NaN、posinf和neginf指定的值。
neg	返回一个带有输入元素负数的新张量。
negative	torch.neg()别名
nextafter	返回输入到其他元素后的下一个浮点值。
polygamma	torch.special.polygamma()别名
positive	返回输入。如果输入是布尔张量，则引发运行时错误。
pow	以指数形式获取输入中每个元素的幂，并返回结果的张量。
quantized_batch_norm	对4D（NCHW）量化张量应用批量标准化。
quantized_max_pool1d	在由多个输入平面组成的输入量化张量上应用1D max池化。
quantized_max_pool2d	在由多个输入平面组成的输入量化张量上应用2D max池化。
rad2deg	返回一个新的张量，其中输入的每个元素都从弧度角度转换为度。
real	返回包含自张量实值的新张量。
reciprocal	返回具有输入元素倒数的新张量
remainder	与torch.fmod() 操作类似，对浮点张量应用C++的std::fmod，对整型张量应用取余运算
round	返回一个新的张量，将输入的每个元素四舍五入为最接近的整数。
rsqrt	返回一个新的张量，该张量的倒数为每个输入元素的平方根。
sigmoid	torch.special.expit()别名
sign	返回带有输入元素符号的新张量。
sgn	对复数张量运用torch.sign()类似功能
signbit	测试输入的每个元素是否设置了符号位（小于零）。
sin	返回带有输入元素正弦的新张量。
sinc	torch.special.sinc()别名
sinh	返回具有输入元素双曲正弦的新张量。
sqrt	返回带有输入元素平方根的新张量。
square	返回一个带有输入元素平方的新张量。
sub	从输入中减去按alpha缩放的其他值。
subtract	torch.sub()别名
tan	返回一个与输入元素相切的新张量。
tanh	返回具有输入元素双曲正切的新张量。
true_divide	torch.div()别名，设置rounding_mode=None.
trunc	返回一个新的张量，其中包含输入元素的截断整数值。
xlogy	torch.special.xlogy()别名

归纳操作

argmax	返回输入张量中所有元素的最大值的索引。
argmin	返回展平张量或沿维度的最小值的索引
amax	返回给定维度中每个输入张量切片的最大值。
amin	返回给定维度中输入张量的每个切片的最小值。
aminmax	计算输入张量的最小值和最大值。
all	测试输入中的所有元素是否计算为True。
any	测试输入中的所有元素是否计算为True。
max	返回输入张量中所有元素的最大值。
min	返回输入张量中所有元素的最小值。
dist	返回(input - other)p-范数
logsumexp	返回给定维度中输入张量每行的求和指数的对数。
mean	返回输入张量中所有元素的平均值。
nanmean	计算指定维度上所有非NaN元素的平均值。
median	返回输入值的中值。
nanmedian	返回输入值的中值，忽略NaN值。
mode	返回一个namedtuple（值，索引），其中值是给定维度dim中输入张量每行的模式值，即最常出现在该行中的值，索引是找到的每个模式值的索引位置。
norm	返回给定张量的矩阵范数或向量范数。
nansum	返回所有元素的和，不将数字（NaN）视为零。
prod	返回输入张量中所有元素的乘积。
quantile	计算沿维度dim的每一行输入张量的第q个分位数。
nanquantile	计算分位数q，忽略输入中NaN值。
std	如果无偏为真，将使用贝塞尔修正。
std_mean	如果无偏为真，贝塞尔校正将用于计算标准偏差。
sum	返回输入张量中所有元素的总和。
unique	返回输入张量的唯一元素。
unique_consecutive	从每个连续的等效元素组中消除除第一个元素以外的所有元素。
var	如果无偏为真，将使用贝塞尔修正。
var_mean	如果无偏为真，贝塞尔校正将用于计算方差。
count_nonzero	计算张量输入中沿给定dim的非零值数。

比较操作

allclose	此函数检查所有输入和其他输入是否满足以下条件：∣input−other∣≤atol+rtol×∣other∣
argsort	返回按值升序对给定维度上的张量进行排序的索引。
eq	计算元素相等
equal	如果两个张量的大小和元素相同，则为True，否则为False。
ge	计算input大于等于other的元素
greater_equal	torch.ge()别名
gt	计算input>other的元素
greater	torch.gt()别名
isclose	返回一个带有布尔元素的新张量，表示输入的每个元素是否与另一个元素的对应元素“接近”。
isfinite	返回一个新的张量，其中包含表示每个元素是否有限的布尔元素。
isin	测试元素的每个元素是否在测试元素中。
isinf	测试输入的每个元素是否为无穷大（正无穷大或负无穷大）。
isposinf	测试输入的每个元素是否为正无穷大。
isneginf	测试输入的每个元素是否为负无穷大。
isnan	返回一个新的张量，其中布尔元素表示输入的每个元素是否为NaN。
isreal	返回一个新的张量，其中布尔元素表示输入的每个元素是否为实值。
kthvalue	返回一个namedtuple（值、索引），其中值是给定维度中输入张量每行的第k个最小元素。
le	计算input小于等于other 的元素
less_equal	torch.le()别名
lt	计算input<other的元素
less	torch.lt()别名
maximum	计算输入和其他元素的元素最大值。
minimum	计算输入和其他元素的元素最小值。
fmax	计算输入和其他元素的元素最大值。
fmin	计算输入和其他元素的元素最小值。
ne	计算input不等号other的元素.
not_equal	torch.ne()别名
sort	按值按升序将输入张量的元素沿给定维度排序。
topk	返回给定输入张量沿给定维度的k个最大元素。
msort	将输入张量的元素沿其第一维度按值升序排序。

频谱操作

stft	短时傅里叶变换 (STFT).
istft	短时傅里叶逆变换。
bartlett_window	Bartlett窗口函数。
blackman_window	Blackman窗口函数。
hamming_window	汉明窗口函数。
hann_window	汉恩窗口功能。
kaiser_window	使用窗口长度窗口长度和形状参数beta计算Kaiser窗口。

其他操作

atleast_1d	返回每个零维输入张量的一维视图。
atleast_2d	返回每个零维输入张量的二维视图。
atleast_3d	返回每个零维输入张量的三维视图。
bincount	计算非负整数数组中每个值的频率。
block_diag	根据提供的张量创建块对角矩阵。
broadcast_tensors	​ 根据广播语义广播给定的张量。
broadcast_to	将输入广播到形状。
broadcast_shapes	broadcast_tensors() 类似，但对形状进行操作
bucketize	返回输入中每个值所属的存储桶的索引，其中存储桶的边界由边界设置。
cartesian_prod	求给定张量序列的笛卡尔积。
cdist	批量计算两个行向量集合的每对之间的p-范数距离。
clone	返回输入的副本。
combinations	计算给定张量的长度r的组合。
corrcoef	估计输入矩阵给出的变量的皮尔逊积矩相关系数矩阵，其中行是变量，列是观察值。
cov	估计输入矩阵给出的变量的协方差矩阵，其中行是变量，列是观测值。
cross	返回输入和其他维度中向量的叉积。
cummax	返回namedtuple（值、索引），其中值是维度dim中输入元素的累积最大值。
cummin	返回namedtuple（值、索引），其中值是维度dim中输入元素的累积最小值。
cumprod	返回维度dim中输入元素的累积乘积。
cumsum	返回维度dim中输入元素的累积和。
diag	如果输入是一个向量（1-D张量），则返回一个以输入元素为对角线的2-D平方张量。 如果输入是一个矩阵（二维张量），则返回一个包含输入对角线元素的一维张量。
diag_embed	创建一个张量，其特定二维平面（由dim1和dim2指定）的对角线由输入填充。
diagflat	如果输入是一个向量（1-D张量），则返回一个以输入元素为对角线的2-D平方张量。 如果输入是一个具有多个维度的张量，则返回一个对角线元素等于平坦输入的二维张量。
diagonal	返回输入的局部视图，其相对于dim1和dim2的对角线元素作为标注附加在形状的末尾。
diff	计算给定维度上的第n个正向差。
einsum	使用基于爱因斯坦求和约定的符号，将输入操作数的元素沿指定的维数求和。
flatten	通过将输入重塑为一维张量，将其展平。
flip	逆n-D张量沿给定轴的顺序（以dims为单位）。
fliplr	在左/右方向翻转张量，返回一个新的张量。
flipud	向上/向下翻转张量，返回一个新的张量。
kron	计算输入和其他的Kron乘积。
rot90	在dims轴指定的平面上将n-D张量旋转90度。
gcd	计算输入和其他元素的元素最大公约数（GCD）。
histc	计算张量的直方图。
histogram	计算张量值的直方图。
meshgrid	创建由1D输入指定的坐标网格.
lcm	计算输入和其他元素的元素最小公倍数（LCM）。
logcumsumexp	返回维度dim中输入元素的指数累加和的对数。
ravel	返回一个连续的展平张量。
renorm	返回一个张量，其中沿维度dim输入的每个子张量都被规格化，使得子张量的p-范数低于值maxnorm
repeat_interleave	重复张量的元素。
roll	沿着给定的维度滚动张量。
searchsorted	从排序的_序列的最内层维度查找索引，这样，如果在索引之前插入值中的对应值，则排序的_序列中对应的最内层维度的顺序将保持不变。
tensordot	返回a和b在多个维度上的收缩。
trace	返回输入二维矩阵对角线元素的总和。
tril	返回矩阵的下三角部分（二维张量）或一批矩阵输入，结果张量输出的其他元素设置为0。
tril_indices	返回2×N张量中逐列矩阵下三角部分的索引，其中第一行包含所有索引的行坐标，第二行包含列坐标。
triu	返回矩阵的上三角部分（二维张量）或一批矩阵输入，结果张量输出的其他元素设置为0。
triu_indices	返回2×N张量中逐列矩阵上三角部分的索引，其中第一行包含所有索引的行坐标，第二行包含列坐标。
vander	生成一个范德蒙矩阵。
view_as_real	以实张量形式返回输入视图。
view_as_complex	以复数张量形式返回输入视图。
resolve_conj	如果输入的共轭位设置为True，则返回具有物化共轭的新张量，否则返回输入。
resolve_neg	如果输入的负位设置为True，则返回带有物化否定的新张量，否则返回输入。

BLAS和LAPACK操作

addbmm	执行batch1和batch2中存储的矩阵的批量矩阵积，减少了加法步骤（所有矩阵乘法沿第一维度累积）。
addmm	执行矩阵mat1和mat2的矩阵乘法。
addmv	执行矩阵mat和向量vec的矩阵向量积。
addr	执行向量vec1和vec2的外积，并将其添加到矩阵输入中。
baddbmm	执行batch1和batch2中矩阵的批量矩阵积。
bmm	执行输入和mat2中存储的矩阵的批量矩阵积。
chain_matmul	返回NN二维张量的矩阵积。
cholesky	计算对称正定矩阵AA或对称正定矩阵批次的Cholesky分解。
cholesky_inverse	计算对称正定矩阵AA的逆矩阵，使用其Cholesky因子uu：返回矩阵inv。
cholesky_solve	求解一个线性方程组，其中一个半正定矩阵在给定其Cholesky因子矩阵uu的情况下被求逆。
dot	计算两个一维张量的点积。
eig	计算实方阵的特征值和特征向量。
geqrf	这是一个用于直接调用LAPACK的geqrf的低级函数。
ger	torch.outer()别名
inner	计算一维张量的点积。
inverse	torch.linalg.inv()别名
det	torch.linalg.det()别名
logdet	计算一个或多个方阵的对数行列式。
slogdet	torch.linalg.slogdet()别名
lstsq	计算大小为（m×n）的满秩矩阵AA和大小为（m×k）的矩阵BB的最小二乘和最小范数问题的解。
lu	计算一个或多个矩阵a的LU分解。
lu_solve	使用从torch.lu()得到的A的部分旋转LU分解返回线性系统Ax=b的LU解
lu_unpack	将数据解包，并从张量的LU分解为张量L和U，以及置换张量P，这样LU_数据，LU_pivots = (P @ L @ U).lu().
matmul	两个张量的矩阵积。
matrix_power	torch.linalg.matrix_power()别名
matrix_rank	返回二维张量的数值秩。
matrix_exp	计算一个方阵或一批中每个方阵的矩阵指数。
mm	对矩阵输入和mat2执行矩阵乘法。
mv	执行矩阵输入和向量向量向量的矩阵向量积。
orgqr	torch.linalg.householder_product()别名
ormqr	计算Householder矩阵与一般矩阵乘积的矩阵乘法。
outer	输入和向量2的外积。
pinverse	torch.linalg.pinv()别名
qr	计算一个矩阵或一批矩阵输入的QR分解，并返回一个命名的张量（Q，R），使得input=QR，其中Q是正交矩阵或正交矩阵的一批，R是上三角矩阵或上三角矩阵的一批。
solve	此函数返回由AX=B表示的线性方程组的解和A的LU分解，以作为命名的对偶解LU。
svd	计算一个矩阵或一批矩阵输入的奇异值分解。
svd_lowrank	返回一个矩阵、一批矩阵或一个稀疏矩阵的奇异值分解（U、S、V），使A≈Q dial(S)VT。
pca_lowrank	对低秩矩阵、此类矩阵的批次或稀疏矩阵执行线性主成分分析（PCA）。
symeig	此函数返回实对称或复厄米矩阵输入或其一批输入的特征值和特征向量，用一个命名的偶（特征值、特征向量）表示。
lobpcg	用无矩阵LOBPCG方法求对称正定义广义特征值问题的k个最大（或最小）特征值和相应的特征向量。
trapz	torch.trapezoid()别名
trapezoid	​ 沿dim计算梯形规则。
cumulative_trapezoid	沿dim累积计算梯形规则。
triangular_solve	用三角形系数矩阵A和多个右手边b求解方程组。
vdot	计算两个一维张量的点积。