PyTorch 张量的加权平均值

Question

我有两个形式为 [y11, y12] 和 [y21, y22] 的 Pytorch 张量。如何获得两个张量的加权平均值？

Answer 1

您可以使用添加两个张量，然后使用 假设张量 1 的权重为 0.6，张量 2 的权重为 0.4 来torch.add获取输出张量的平均值，例如：torch.mean

tensor1 = [y11, y12] * 0.6 # multiplying with weight
tensor2 = [y21, y22] * 0.4 # multiplying with weight
pt_addition_result_ex = tensor1.add(tensor2) # addition of two tensors

torch.mean(pt_addition_result_ex) # mean of output tensors