Add files so that we can merge. VGG state is broken, since it needs normalization, but we need to all use the CMake system.

Author: Iainmon

examples/torch_model_loading/torch_load.chpl

@@ -1,9 +1,15 @@
 use Tensor;
 
 
-proc main() {
-    const a = ndarray.loadPyTorchTensorDictWithKey(2,"models/my_tensor_dict.pt","a");
-    const b = ndarray.loadPyTorchTensorDictWithKey(2,"models/my_tensor_dict.pt","b");
-    writeln("a sum: ", a.sum());
-    writeln("b sum: ", b.sum());
+proc main(args: [] string) {
+    // const a = ndarray.loadPyTorchTensorDictWithKey(2,"models/my_tensor_dict.pt","a");
+    // const b = ndarray.loadPyTorchTensorDictWithKey(2,"models/my_tensor_dict.pt","b");
+    // writeln("a sum: ", a.sum());
+    // writeln("b sum: ", b.sum());
+
+    var image = ndarray.loadFrom(args[1],3,real(32));
+    writeln("Loaded image: ", args[1]);
+    writeln("Image shape: ", image.shape);
+
+    image.saveImage("test.jpg");
 }

lib/NDArray.chpl

@@ -2244,7 +2244,7 @@ proc type ndarray.loadImage(imagePath: string, type eltType = defaultEltType): n
     return img;
 }
 
-proc ref ndarray.saveImage(imagePath: string) where rank == 3 {
+proc ref ndarray.saveImage(imagePath: string) throws where rank == 3 {
 
     // compilerWarning("I have not implemented ndarray.saveImage");
     import Image;