PyTorch:addmm功能的目的(PyTorch: purpose of addmm funct

2019-09-25 15:30发布

什么是以下PyTorch功能(目的DOC ):

torch.addmm(beta=1, mat, alpha=1, mat1, mat2, out=None)

更具体地说,是没有任何理由喜欢这个功能,而不是仅仅使用

beta * mat + alpha * (mat1 @ mat2)

Answer 1:

addmm功能是方程的优化版本beta*mat + alpha*(mat1 @ mat2) 我跑了一些测试,并定时它们的执行。

  • 如果beta=1, alpha=1 ,则这两个语句(执行addmm和手册)是大致相同的( addmm仅仅是一个快一点),而不管矩阵的大小。

  • 如果betaalpha不为1,则addmm比手动执行对于较小的矩阵快两倍(总元件在10 5顺序)。 但是,如果矩阵是大的(在10 6阶),加速似乎可以忽略不计( 39ms V / S 41ms



文章来源: PyTorch: purpose of addmm function
标签: pytorch