as_torch_optimizer function

Convert to TorchOptimizer