代码之家  ›  专栏  ›  技术社区  ›  iago-lito

外和等

  •  0
  • iago-lito  · 技术社区  · 6 年前

    Numpy为任何应用程序提供优化的外部操作 RxR -> R np.multiply.outer np.subtract.outer ,表现为:

    >>> np.subtract.outer([6, 5, 4], [3, 2, 1])
    array([[3, 4, 5],
           [2, 3, 4],
           [1, 2, 3]])
    

    Pytorch

    1 回复  |  直到 6 年前
        1
  •  5
  •   akuiper    6 年前

    根据 documenation :

    许多PyTorch操作支持NumPy广播语义。

    x = torch.Tensor([6, 5, 4])
    y = torch.Tensor([3, 2, 1])
    
    x.reshape(-1, 1) - y
    #tensor([[3., 4., 5.],
    #        [2., 3., 4.],
    #        [1., 2., 3.]])