compiled-inference
compiled-inference copied to clipboard
fixed a bug
This torch.max doesn't take keepdim argument.