Class Activation Mapping(CAM) In Android

I could draw the CAM using this tutorial
Class Activation Mapping In PyTorch · Ian Pointer (snappishproductions.com)
, but I was wondering if I can draw CAM for predicted image in android.