昇腾社区首页
中文
注册

类说明

Linear是一个自定义的线性层,类似于PyTorch中的torch.nn.Linear,初始化参数新增op_type。实现了矩阵乘法的核心功能,并支持不同的操作类型(op_type)以及对输入张量的处理。该函数在MindIE SD仓的路径为:mindiesd/layers/linear.py,其具体使用方法请参见Linear

成员

成员名称

描述

__init__

类初始化函数。

forward

根据不同的op_type设置,执行线性变换。