Merge pull request #57 from Etienne66/master

Fix issue #47

Author: sniklaus

correlation/correlation.py

@@ -246,9 +246,11 @@ def cupy_kernel(strFunction, objVariables):
 
         strTensor = objMatch.group(4)
         intSizes = objVariables[strTensor].size()
-
-        strKernel = strKernel.replace(objMatch.group(), str(intSizes[intArg]))
-    # end
+        if torch.is_tensor(intSizes[intArg]):
+            strKernel = strKernel.replace(objectMatch.group(), str(intSizes[intArg].item()))
+        else:
+            strKernel = strKernel.replace(objectMatch.group(), str(intSizes[intArg]))
+        # end
 
     while True:
         objMatch = re.search('(VALUE_)([0-4])(\()([^\)]+)(\))', strKernel)
@@ -394,4 +396,4 @@ def __init__(self):
     def forward(self, tenOne, tenTwo):
         return _FunctionCorrelation.apply(tenOne, tenTwo)
     # end
-# end
+# end