类说明
Linear是一个自定义的线性层,类似于PyTorch中的torch.nn.Linear,初始化参数新增op_type。实现了矩阵乘法的核心功能,并支持不同的操作类型(op_type)以及对输入张量的处理。该函数在MindIE SD仓的路径为:mindiesd/layers/linear.py,其具体使用方法请参见Linear。
成员
成员名称 |
描述 |
---|---|
__init__ |
类初始化函数。 |
forward |
根据不同的op_type设置,执行线性变换。 |
父主题: class Linear