Recently, I am reading the paper Input convex neural network whose implementation is using tensorflow. I am working to reproducing the network and do some simulation using PyTorch.
But how to set the constraints of the weights of the network.