Средневзвешенное значение тензоров PyTorch

вы можете добавить два тензора, используя torch.add, а затем получить среднее значение выходного тензора, используя torch.mean, предполагая, что вес равен 0,6 для tensor1 и 0,4 для примера tensor2:

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