操作

abs(a, /, *[, stream])	逐元素的绝对值。
add(a, b[, stream])	逐元素相加。
addmm(c, a, b, /[, alpha, beta, stream])	矩阵乘法与加法及可选缩放。
all(a, /[, axis, keepdims, stream])	对给定轴进行与操作。
allclose(a, b, /[, rtol, atol, equal_nan, ...])	两个数组的近似比较。
any(a, /[, axis, keepdims, stream])	在给定轴上进行或归约。
arange(-> array)	重载函数。
arccos(a, /, *[, stream])	逐元素反余弦。
arccosh(a, /, *[, stream])	逐元素的反双曲余弦。
arcsin(a, /, *[, stream])	逐元素的反弦函数。
arcsinh(a, /, *[, stream])	逐元素反双曲正弦。
arctan(a, /, *[, stream])	逐元素反正切。
arctan2(a, b, /, *[, stream])	两个数组比值的逐元素反正切。
arctanh(a, /, *[, stream])	逐元素反双曲正切。
argmax(a, /[, axis, keepdims, stream])	沿轴的最大值的索引。
argmin(a, /[, axis, keepdims, stream])	沿轴的最小值的索引。
argpartition(a, /, kth[, axis, stream])	返回分区数组的索引。
argsort(a, /[, axis, stream])	返回排序数组的索引。
array_equal(a, b[, equal_nan, stream])	数组相等性检查。
as_strided(a, /[, shape, strides, offset, ...])	使用给定的形状和步幅创建数组的视图。
atleast_1d(*arys[, stream])	将所有数组转换为至少有一个维度。
atleast_2d(*arys[, stream])	将所有数组转换为至少有两个维度。
atleast_3d(*arys[, stream])	将所有数组转换为至少具有三个维度。
bitwise_and(a, b[, stream])	逐元素的位与操作。
bitwise_invert(a[, stream])	逐元素的位取反。
bitwise_or(a, b[, stream])	逐元素的按位或。
bitwise_xor(a, b[, stream])	逐元素的按位异或。
block_masked_mm(a, b, /[, block_size, ...])	带有块掩码的矩阵乘法。
broadcast_to(a, /, shape, *[, stream])	将数组广播到给定的形状。
ceil(a, /, *[, stream])	逐元素向上取整。
clip(a, /, a_min, a_max, *[, stream])	将数组的值限制在给定的最小值和最大值之间。
concatenate(arrays[, axis, stream])	沿给定轴连接数组。
conj(a, *[, stream])	返回输入的逐元素复共轭。
conjugate(a, *[, stream])	返回输入的元素级复共轭。
convolve(a, v, /[, mode, stream])	一维数组的离散卷积。
conv1d(input, weight, /[, stride, padding, ...])	对具有多个通道的输入进行一维卷积
conv2d(input, weight, /[, stride, padding, ...])	对具有多个通道的输入进行2D卷积
conv3d(input, weight, /[, stride, padding, ...])	对具有多个通道的输入进行3D卷积
conv_transpose1d(input, weight, /[, stride, ...])	对具有多个通道的输入进行1D转置卷积
conv_transpose2d(input, weight, /[, stride, ...])	对具有多个通道的输入进行2D转置卷积
conv_transpose3d(input, weight, /[, stride, ...])	对具有多个通道的输入进行3D转置卷积
conv_general(input, weight, /[, stride, ...])	对具有多个通道的输入进行通用卷积
cos(a, /, *[, stream])	逐元素余弦。
cosh(a, /, *[, stream])	逐元素双曲余弦。
cummax(a, /[, axis, reverse, inclusive, stream])	返回沿给定轴的元素累积最大值。
cummin(a, /[, axis, reverse, inclusive, stream])	返回沿给定轴的元素累积最小值。
cumprod(a, /[, axis, reverse, inclusive, stream])	返回沿给定轴元素的累积乘积。
cumsum(a, /[, axis, reverse, inclusive, stream])	返回沿给定轴的元素累积和。
degrees(a, /, *[, stream])	将角度从弧度转换为度。
dequantize(w, /, scales, biases[, ...])	使用提供的scales和biases以及group_size和bits配置对矩阵w进行反量化。
diag(a, /[, k, stream])	提取对角线或构造对角矩阵。
diagonal(a[, offset, axis1, axis2, stream])	返回指定的对角线。
divide(a, b[, stream])	逐元素除法。
divmod(a, b[, stream])	逐元素的商和余数。
einsum(subscripts, *operands[, stream])	对操作数执行爱因斯坦求和约定。
einsum_path(subscripts, *operands)	计算给定爱因斯坦求和约定的收缩顺序。
equal(a, b[, stream])	元素级别的相等性。
erf(a, /, *[, stream])	逐元素的误差函数。
erfinv(a, /, *[, stream])	逐元素计算erf()的逆函数。
exp(a, /, *[, stream])	逐元素指数。
expm1(a, /, *[, stream])	逐元素的指数减1。
expand_dims(a, /, axis, *[, stream])	在给定的轴上添加一个大小为1的维度。
eye(n[, m, k, dtype, stream])	创建一个单位矩阵或一个一般的对角矩阵。
flatten(a, /[, start_axis, end_axis, stream])	展平数组。
floor(a, /, *[, stream])	逐元素取整。
floor_divide(a, b[, stream])	逐元素整数除法。
full(shape, vals[, dtype, stream])	使用给定值构造一个数组。
gather_mm(a, b, /, lhs_indices, rhs_indices, *)	带有矩阵级收集的矩阵乘法。
gather_qmm(x, w, /, scales, biases[, ...])	执行带有矩阵级收集的量化矩阵乘法。
greater(a, b[, stream])	逐元素大于。
greater_equal(a, b[, stream])	逐元素大于或等于。
hadamard_transform(a[, scale, stream])	沿最终轴执行沃尔什-哈达玛变换。
identity(n[, dtype, stream])	创建一个单位方阵。
imag(a, /, *[, stream])	返回复数数组的虚部。
inner(a, b, /, *[, stream])	对于一维数组的向量普通内积，在更高维度上是对最后轴的和积。
isfinite(a[, stream])	返回一个布尔数组，指示哪些元素是有限的。
isclose(a, b, /[, rtol, atol, equal_nan, stream])	返回一个布尔数组，其中两个数组在容差范围内逐元素相等。
isinf(a[, stream])	返回一个布尔数组，指示哪些元素是正负无穷大。
isnan(a[, stream])	返回一个布尔数组，指示哪些元素是NaN。
isneginf(a[, stream])	返回一个布尔数组，指示哪些元素是负无穷大。
isposinf(a[, stream])	返回一个布尔数组，指示哪些元素是正无穷大。
issubdtype(arg1, arg2)	检查一个Dtype或DtypeCategory是否是另一个的子类型。
kron(a, b, *[, stream])	计算两个数组 a 和 b 的克罗内克积。
left_shift(a, b[, stream])	逐元素左移。
less(a, b[, stream])	逐元素小于。
less_equal(a, b[, stream])	逐元素小于或等于。
linspace(start, stop[, num, dtype, stream])	在区间 [start, stop] 上生成 num 个均匀分布的数字。
load(file, /[, format, return_metadata, stream])	从二进制文件加载数组。
log(a, /, *[, stream])	逐元素的自然对数。
log2(a, /, *[, stream])	逐元素的以2为底的对数。
log10(a, /, *[, stream])	逐元素的以10为底的对数。
log1p(a, /, *[, stream])	数组逐元素加一后的自然对数。
logaddexp(a, b, /, *[, stream])	逐元素的log-add-exp。
logical_not(a, /, *[, stream])	逐元素逻辑非。
logical_and(a, b, /, *[, stream])	逐元素的逻辑与。
logical_or(a, b, /, *[, stream])	逐元素逻辑或。
logsumexp(a, /[, axis, keepdims, stream])	对给定轴进行log-sum-exp缩减。
matmul(a, b, /, *[, stream])	矩阵乘法。
max(a, /[, axis, keepdims, stream])	在给定轴上进行 max 归约。
maximum(a, b, /, *[, stream])	逐元素最大值。
mean(a, /[, axis, keepdims, stream])	计算给定轴上的平均值。
meshgrid(*arrays[, sparse, indexing, stream])	从一维坐标数组生成多维坐标网格
min(a, /[, axis, keepdims, stream])	在给定轴上进行min缩减。
minimum(a, b, /, *[, stream])	逐元素最小值。
moveaxis(a, /, source, destination, *[, stream])	将轴移动到新位置。
multiply(a, b[, stream])	逐元素乘法。
nan_to_num(a[, nan, posinf, neginf, stream])	用有限数字替换NaN和Inf值。
negative(a, /, *[, stream])	逐元素取反。
not_equal(a, b[, stream])	逐元素不相等。
ones(shape[, dtype, stream])	构造一个全为1的数组。
ones_like(a, /, *[, stream])	一个与输入数组形状相同的全1数组。
outer(a, b, /, *[, stream])	计算两个一维数组的外积，如果传入的数组不是一维的，将先执行展平操作。
partition(a, /, kth[, axis, stream])	返回数组的分区副本，使得较小的kth元素排在前面。
pad(a, pad_width[, mode, constant_values, ...])	用常数值填充数组
power(a, b, /, *[, stream])	逐元素的幂运算。
prod(a, /[, axis, keepdims, stream])	在给定轴上进行乘积缩减。
put_along_axis(a, /, indices, values[, ...])	在指定索引处沿轴放置值。
quantize(w, /[, group_size, bits, stream])	使用每个元素bits位对矩阵w进行量化。
quantized_matmul(x, w, /, scales, biases[, ...])	使用量化矩阵 w 执行矩阵乘法。
radians(a, /, *[, stream])	将角度从度转换为弧度。
real(a, /, *[, stream])	返回复数数组的实部。
reciprocal(a, /, *[, stream])	逐元素倒数。
remainder(a, b[, stream])	逐元素的除法余数。
repeat(array, repeats[, axis, stream])	沿指定轴重复数组。
reshape(a, /, shape, *[, stream])	在保持大小不变的情况下重塑数组。
right_shift(a, b[, stream])	逐元素右移。
roll(a, shift[, axis, stream])	沿给定轴滚动数组元素。
round(a, /[, decimals, stream])	四舍五入到指定的小数位数。
rsqrt(a, /, *[, stream])	逐元素的倒数和平方根。
save(file, arr)	将数组保存为.npy格式的二进制文件。
savez(file, *args, **kwargs)	将多个数组保存为未压缩的.npz格式的二进制文件。
savez_compressed(file, *args, **kwargs)	将多个数组保存为压缩的.npz格式的二进制文件。
save_gguf(file, arrays, metadata)	将数组保存为.gguf格式的二进制文件。
save_safetensors(file, arrays[, metadata])	将数组保存为.safetensors格式的二进制文件。
sigmoid(a, /, *[, stream])	逐元素的逻辑S形函数。
sign(a, /, *[, stream])	逐元素的符号。
sin(a, /, *[, stream])	逐元素正弦。
sinh(a, /, *[, stream])	逐元素双曲正弦。
slice(a, start_indices, axes, slice_size, *)	从输入数组中提取一个子数组。
slice_update(a, update, start_indices, axes, *)	更新输入数组的子数组。
softmax(a, /[, axis, stream])	沿给定轴执行softmax。
sort(a, /[, axis, stream])	返回数组的排序副本。
split(a, /, indices_or_sections[, axis, stream])	沿给定轴分割数组。
sqrt(a, /, *[, stream])	逐元素平方根。
square(a, /, *[, stream])	逐元素平方。
squeeze(a, /[, axis, stream])	从数组中移除长度为1的轴。
stack(arrays[, axis, stream])	将数组沿新轴堆叠。
std(a, /[, axis, keepdims, ddof, stream])	计算给定轴上的标准差。
stop_gradient(a, /, *[, stream])	停止计算梯度。
subtract(a, b[, stream])	逐元素减法。
sum(a, /[, axis, keepdims, stream])	在给定的轴上对数组进行求和归约。
swapaxes(a, /, axis1, axis2, *[, stream])	交换数组的两个轴。
take(a, /, indices[, axis, stream])	沿轴取元素。
take_along_axis(a, /, indices[, axis, stream])	沿着指定轴在指定索引处取值。
tan(a, /, *[, stream])	逐元素的正切。
tanh(a, /, *[, stream])	逐元素双曲正切。
tensordot(a, b, /[, axes, stream])	沿着指定的轴计算张量点积。
tile(a, reps, /, *[, stream])	通过重复a的次数来构造一个数组，次数由reps给出。
topk(a, /, k[, axis, stream])	返回输入中沿给定轴的前k个最大元素。
trace(a, /[, offset, axis1, axis2, dtype, ...])	返回给定数组中沿指定对角线的和。
transpose(a, /[, axes, stream])	转置数组的维度。
tri(n, m, k[, dtype, stream])	一个数组，在给定对角线及其下方为1，其他地方为0。
tril(x, k, *[, stream])	将数组在给定对角线以上的部分置零。
triu(x, k, *[, stream])	将给定对角线以下的数组元素置零。
unflatten(a, /, axis, shape, *[, stream])	将数组的一个轴展开为指定的形状。
var(a, /[, axis, keepdims, ddof, stream])	计算给定轴上的方差。
view(a, dtype[, stream])	将数组视为不同的类型。
where(condition, x, y, /, *[, stream])	根据condition从x或y中选择。
zeros(shape[, dtype, stream])	构造一个全零数组。
zeros_like(a, /, *[, stream])	一个与输入数组形状相同的零数组。