import torch
v1 = torch.arange(1., 5.)
v2 = torch.arange(1., 4.)
print(torch.ger(v1, v2))
tensor([[ 1., 2., 3.],
[ 2., 4., 6.],
[ 3., 6., 9.],
[ 4., 8., 12.]])
微信扫一扫
import torch
v1 = torch.arange(1., 5.)
v2 = torch.arange(1., 4.)
print(torch.ger(v1, v2))
tensor([[ 1., 2., 3.],
[ 2., 4., 6.],
[ 3., 6., 9.],
[ 4., 8., 12.]])
相关推荐