使用点积的热图可视化

您好，我用点积版本的non_local对f进行softmax之后，
        f = torch.matmul(theta_x, phi_x)
        N = f.size(-1)
        f_div_C = f / N
        sft = nn.Softmax(dim=2)
        f_div_C = sft(f_div_C)
在可视化时还需要对NL_MAP进行normalize吗？如果需要，应该如何normalize呢？
现在我的热图结果只有一个box，看不到正常热图颜色的变化，这是什么原因呢？