nndeploy.op.functional

函数形式Op

Functions

abs(input)
acos(input)
add(input1, input2)
asin(input)
atan(input)
batch_norm(input, scale, bias, mean, var[, ...])
ceil(input)
concat(input1, input2[, axis])
conv(input, weight[, bias, stride, padding, ...])
cos(input)
cosh(input)
dequantize_linear(input, scale, zero_point)
erf(input)
exp(input)
flatten(input[, axis])
floor(input)
gather(input, index[, axis])
gemm(input_a, input_b[, input_c, alpha, ...])
global_averagepool(input)
hardsigmoid(input)
log(input)
mat_mul(input1, input2[, bias, transposeA, ...])
maxpool(input, kernel_size[, stride, ...])
mul(input1, input2)
qlinear_conv(x, x_scale, x_zero_point, w, ...)
quantize_linear(input, scale, zero_point[, ...])
reciprocal(input)
relu(input)
reshape(input, shape[, allowzero])
rms_norm(input, weight[, epsilon])
round(input)
selu(input)
sigmoid(input)
sign(input)
sin(input)
sinh(input)
slice(input, starts, ends, axes, steps)
softmax(input[, axis])
split(input[, section, num_outputs, axis])	Python 封装的 Split 接口 支持两种互斥模式： 1) num_outputs：等分或最后一块变小 2) section：list/tuple/int 指定每段长度（自动转为 1-D int64 Tensor） ;具有更高优先级
sqrt(input)
tan(input)
tanh(input)
transpose(input, perm_axis)
where(input1, input2, condition)

nndeploy.op.functional.conv(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)

nndeploy.op.functional.concat(input1, input2, axis=0)

nndeploy.op.functional.batch_norm(input, scale, bias, mean, var, epsilon=1e-05)

nndeploy.op.functional.relu(input)

nndeploy.op.functional.hardsigmoid(input)

nndeploy.op.functional.selu(input)

nndeploy.op.functional.tanh(input)

nndeploy.op.functional.abs(input)

nndeploy.op.functional.acos(input)

nndeploy.op.functional.asin(input)

nndeploy.op.functional.atan(input)

nndeploy.op.functional.ceil(input)

nndeploy.op.functional.cos(input)

nndeploy.op.functional.cosh(input)

nndeploy.op.functional.erf(input)

nndeploy.op.functional.exp(input)

nndeploy.op.functional.floor(input)

nndeploy.op.functional.log(input)

nndeploy.op.functional.reciprocal(input)

nndeploy.op.functional.round(input)

nndeploy.op.functional.sign(input)

nndeploy.op.functional.sin(input)

nndeploy.op.functional.sinh(input)

nndeploy.op.functional.sqrt(input)

nndeploy.op.functional.tan(input)

nndeploy.op.functional.add(input1, input2)

nndeploy.op.functional.flatten(input, axis=1)

nndeploy.op.functional.gather(input, index, axis=0)

nndeploy.op.functional.gemm(input_a, input_b, input_c=None, alpha=1.0, beta=1.0, trans_a=0, trans_b=0)

nndeploy.op.functional.global_averagepool(input)

nndeploy.op.functional.maxpool(input, kernel_size, stride=1, padding=0, dilation=1, ceil_mode=False)

nndeploy.op.functional.mat_mul(input1, input2, bias=None, transposeA=False, transposeB=False)

nndeploy.op.functional.mul(input1, input2)

nndeploy.op.functional.rms_norm(input, weight, epsilon=1e-05)

nndeploy.op.functional.reshape(input, shape, allowzero=0)

nndeploy.op.functional.softmax(input, axis=1)

nndeploy.op.functional.slice(input, starts, ends, axes, steps)

nndeploy.op.functional.sigmoid(input)

nndeploy.op.functional.quantize_linear(input, scale, zero_point, axis=1, saturate=True)

nndeploy.op.functional.dequantize_linear(input, scale, zero_point, axis=1)

nndeploy.op.functional.qlinear_conv(x, x_scale, x_zero_point, w, w_scale, w_zero_point, y_scale, y_zero_point, bias=None, stride=1, padding=0, dilation=1, groups=1)

nndeploy.op.functional.where(input1, input2, condition)

nndeploy.op.functional.transpose(input, perm_axis)

nndeploy.op.functional.split(input, section=None, num_outputs=None, axis=0)

Python 封装的 Split 接口 支持两种互斥模式：

1. num_outputs：等分或最后一块变小

2. section：list/tuple/int 指定每段长度（自动转为 1-D int64 Tensor） ;具有更高优先级

返回值：tuple[Tensor, …]