Numpy为任何应用程序提供优化的外部操作 RxR -> R np.multiply.outer 或 np.subtract.outer ,表现为:
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
根据 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.]])