如何找到最大化两个张量内积的概率截止？

Question

我有两个张量：

import torch

target = torch.randint(2, (3,5)) #tensor of 0s & 1s
pred = torch.rand(3, 5) #tensor of prob

# transformed_pred = ?

如何选择截止概率将

pred

转换为 0 和 1 的张量 (

transformed_pred

)，以使

target

和

transformed_pred

之间的点积最大化？

谢谢！

Answer 1

也许我误解了你的问题，但截止概率应该是0.0。 --> 如果您始终选择 1，您的点积将始终等于目标张量中非零元素的总和，即最大值。

还有其他你没有提到的限制吗？