pytorch中的用法

pytorch中的用法

在pytorch1.7版本之后&#就被torch.outer所代替了

下面展示一些 <。

// 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&#sor([2,4,6,8])
>>> v5&#sor([2.,5.,6.,8.])
>>> (v4,v5)
tensor([[ 4., 10., 12., 16.],[ 8., 20., 24., 32.],[12., 30., 36., 48.],[16., 40., 48., 64.]])
>>> (v5,v4)
tensor([[ 4.,  8., 12., 16.],[10., 20., 30., 40.],[12., 24., 36., 48.],[16., 32., 48., 64.]])

我们可以看出来,这个函数是对tensor进行扩维&#(a,b)实际意思是b中的每一个元素乘以a中的元素，进行扩维