blas.py 文件源码

def forward(self, add_vector, matrix, vector):
        self.save_for_backward(matrix, vector)
        output = self._get_output(add_vector)
        return torch.addmv(output, self.alpha, add_vector, self.beta,
                matrix, vector)