目录
元素 *** 作
归纳 *** 作
比较 *** 作
频谱 *** 作
其他 *** 作
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() |
trunc | 返回一个新的张量,其中包含输入元素的截断整数值。 |
xlogy | torch.special.xlogy()别名 |
argmax | 返回输入张量中所有元素的最大值的索引。 |
argmin | 返回展平张量或沿维度的最小值的索引 |
amax | 返回给定维度中每个输入张量切片的最大值。 |
amin | 返回给定维度中输入张量的每个切片的最小值。 |
aminmax | 计算输入张量的最小值和最大值。 |
all | 测试输入中的所有元素是否计算为True。 |
any | 测试输入中的所有元素是否计算为True。 |
max | 返回输入张量中所有元素的最大值。 |
min | 返回输入张量中所有元素的最小值。 |
dist | 返回( |
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 |
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,则返回带有物化否定的新张量,否则返回输入。 |
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_数据, |
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 | 计算两个一维张量的点积。 |
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)