diff --git a/src/pystencils_autodiff/__init__.py b/src/pystencils_autodiff/__init__.py index 7c6f4b9fa2650d3c658af2babb1e1099e7d9632c..fafab5ab22e7ac018d4feeac67ef7dd49f339c4a 100644 --- a/src/pystencils_autodiff/__init__.py +++ b/src/pystencils_autodiff/__init__.py @@ -4,8 +4,8 @@ import pystencils_autodiff.backends # NOQA from pystencils_autodiff._field_to_tensors import ( # NOQA tf_constant_from_field, tf_placeholder_from_field, tf_scalar_variable_from_field, tf_variable_from_field, torch_tensor_from_field) -from pystencils_autodiff.adjoint_field import AdjointField -from pystencils_autodiff.autodiff import ( +from pystencils_autodiff._adjoint_field import AdjointField +from pystencils_autodiff._autodiff import ( AutoDiffAstPair, AutoDiffOp, create_backward_assignments, get_jacobian_of_assignments) __all__ = ['backends', diff --git a/src/pystencils_autodiff/adjoint_field.py b/src/pystencils_autodiff/_adjoint_field.py similarity index 100% rename from src/pystencils_autodiff/adjoint_field.py rename to src/pystencils_autodiff/_adjoint_field.py diff --git a/src/pystencils_autodiff/autodiff.py b/src/pystencils_autodiff/_autodiff.py similarity index 100% rename from src/pystencils_autodiff/autodiff.py rename to src/pystencils_autodiff/_autodiff.py diff --git a/src/pystencils_autodiff/tensorflow.py b/src/pystencils_autodiff/tensorflow.py deleted file mode 100644 index 91a8c76436ec4fd100b8bb72016d6e05832cef5f..0000000000000000000000000000000000000000 --- a/src/pystencils_autodiff/tensorflow.py +++ /dev/null @@ -1,26 +0,0 @@ -import numpy as np - -try: - import tensorflow as tf -except ImportError: - pass - - -def tf_constant_from_field(field, init_val=0): - return tf.constant(init_val, dtype=field.dtype.numpy_dtype, shape=field.shape, name=field.name + '_constant') - - -def tf_scalar_variable_from_field(field, init_val, constraint=None): - var = tf.Variable(init_val, dtype=field.dtype.numpy_dtype, name=field.name + '_variable', constraint=constraint) - return var * tf_constant_from_field(field, 1) - - -def tf_variable_from_field(field, init_val=0, constraint=None): - if isinstance(init_val, (int, float)): - init_val *= np.ones(field.shape, field.dtype.numpy_dtype) - - return tf.Variable(init_val, dtype=field.dtype.numpy_dtype, name=field.name + '_variable', constraint=constraint) - - -def tf_placeholder_from_field(field): - return tf.placeholder(dtype=field.dtype.numpy_dtype, name=field.name + '_placeholder', shape=field.shape) diff --git a/src/pystencils_autodiff/torch.py b/src/pystencils_autodiff/torch.py deleted file mode 100644 index 918f537035d6d51467e6c9ecb9f2f3d32ffcb805..0000000000000000000000000000000000000000 --- a/src/pystencils_autodiff/torch.py +++ /dev/null @@ -1,23 +0,0 @@ -# -*- coding: utf-8 -*- -# -# Copyright © 2019 Stephan Seitz <stephan.seitz@fau.de> -# -# Distributed under terms of the GPLv3 license. - -""" - -""" - -import numpy as np - -try: - import torch -except ImportError: - pass - - -def torch_tensor_from_field(field, init_val=0, cuda=True, requires_grad=False): - if isinstance(init_val, (int, float)): - init_val *= np.ones(field.shape, field.dtype.numpy_dtype) - device = torch.device('cuda' if cuda else 'cpu') - return torch.tensor(init_val, requires_grad=requires_grad, device=device)