Imagine I have a scalar T , this T is gonna be used as a threshold in my network. i.e.
TensorA = torch.where(TensorB > T*Means, Ones, Zeros).
Right now I have T = torch.tensor(1.0), but I want to give it the ability to change and be learnable.
Is the way to do that?
In other words, how can I wrap it in a way to be learnable?