numpy.dot与pytorch.dot的等价性

都对于一维向量时,numpy和pytorch的dot方法等价,都表示两向量内积,当多维时,numpy.dot可以很好的扩展,而torch则需要更换为mm方法

numpy.dot与pytorch.dot的等价性_第1张图片

你可能感兴趣的:(Pytorch)