blas.py 文件源码

def forward(self, add_matrix, vector1, vector2):
        self.save_for_backward(vector1, vector2)
        output = self._get_output(add_matrix)
        return torch.addr(output, self.alpha, add_matrix, self.beta,
                vector1, vector2)