nndeploy.op.expr

Classes

Add()
BatchNorm(scale_name, bias_name, mean_name, ...)
Conv(in_channels, out_channels, kernel_size)
Flatten([axis])
Gemm(weight_name[, bias_name, alpha, beta, ...])
GlobalAveragePool()
MaxPool(kernel_size[, stride, padding, ...])
Relu()
Sigmoid()
Softmax([axis])

class nndeploy.op.expr.Conv(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=[1, 1], groups=1, weight_name='conv.weight', bias_name='conv.bias')

基类：Module

__init__(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=[1, 1], groups=1, weight_name='conv.weight', bias_name='conv.bias')

forward(data)

makeExpr(data)

dump(file_path)

run() → list

setInputs(inputs)

class nndeploy.op.expr.Relu

基类：Module

__init__()

forward(data)

makeExpr(data)

dump(file_path)

run() → list

setInputs(inputs)

class nndeploy.op.expr.Add

基类：Module

__init__()

forward(x, y)

makeExpr(x, y)

dump(file_path)

run() → list

setInputs(inputs)

class nndeploy.op.expr.BatchNorm(scale_name, bias_name, mean_name, var_name, eps=1e-05, momentum=0.9)

基类：Module

__init__(scale_name, bias_name, mean_name, var_name, eps=1e-05, momentum=0.9)

forward(data)

makeExpr(data)

dump(file_path)

run() → list

setInputs(inputs)

class nndeploy.op.expr.Softmax(axis=-1)

基类：Module

__init__(axis=-1)

forward(data)

makeExpr(data)

dump(file_path)

run() → list

setInputs(inputs)

class nndeploy.op.expr.Sigmoid

基类：Module

__init__()

forward(data)

makeExpr(data)

dump(file_path)

run() → list

setInputs(inputs)

class nndeploy.op.expr.Gemm(weight_name, bias_name=None, alpha=1.0, beta=1.0, trans_a=0, trans_b=0)

基类：Module

__init__(weight_name, bias_name=None, alpha=1.0, beta=1.0, trans_a=0, trans_b=0)

forward(data)

makeExpr(data)

dump(file_path)

run() → list

setInputs(inputs)

class nndeploy.op.expr.Flatten(axis=1)

基类：Module

__init__(axis=1)

forward(data)

makeExpr(data)

dump(file_path)

run() → list

setInputs(inputs)

class nndeploy.op.expr.MaxPool(kernel_size, stride=1, padding=0, dilation=1, ceil_mode=False)

基类：Module

__init__(kernel_size, stride=1, padding=0, dilation=1, ceil_mode=False)

forward(data)

makeExpr(data)

dump(file_path)

run() → list

setInputs(inputs)

class nndeploy.op.expr.GlobalAveragePool

基类：Module

__init__()

forward(data)

makeExpr(data)

dump(file_path)

run() → list

setInputs(inputs)