本文主要是介绍pytorch中torch.ger的用法,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
在pytorch1.7版本之后,torch.ger就被torch.outer所代替了
下面展示一些 torch.ger
。
// An highlighted block
>>> import torch
>>> v1 = torch.arange(1., 5.)
>>> v2 = torch.arange(1., 4.)
>>> torch.outer(v1, v2)
tensor([[ 1., 2., 3.],[ 2., 4., 6.],[ 3., 6., 9.],[ 4., 8., 12.]])
>>> v4=torch.tensor([2,4,6,8])
>>> v5=torch.tensor([2.,5.,6.,8.])
>>> torch.ger(v4,v5)
tensor([[ 4., 10., 12., 16.],[ 8., 20., 24., 32.],[12., 30., 36., 48.],[16., 40., 48., 64.]])
>>> torch.ger(v5,v4)
tensor([[ 4., 8., 12., 16.],[10., 20., 30., 40.],[12., 24., 36., 48.],[16., 32., 48., 64.]])
我们可以看出来,torch.ger这个函数是对tensor进行扩维,torch.ger(a,b)实际意思是b中的每一个元素乘以a中的元素,进行扩维
这篇关于pytorch中torch.ger的用法的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!