Remove print stale statements
@@ -73,12 +73,10 @@ class Tensor(JavaObjectWrapper):
@staticmethod
def _to_ndarray(java_tensor: JavaObject, primitive: bool = False) -> Any:
- print(primitive)
if java_tensor.getRank() == 0:
if primitive == True:
return np.array(java_tensor.scalar()).item()
else:
- print("scalar")
return np.array(java_tensor.scalar())
else:
return np.array(list(java_tensor.asFlatArray())).reshape(java_tensor.getShape())
