def setup_backends():
if jax:
from jax.config import config
config.update("jax_enable_x64", True)
if tf:
from tensorflow.python.ops.numpy_ops import np_config
np_config.enable_numpy_behavior()
def testMethods(self):
a = tf.constant(1.)
for name in {
'T', 'astype', 'ravel', 'transpose', 'reshape', 'clip', 'size',
'tolist'
}:
with self.assertRaisesRegex(AttributeError,
'enable_numpy_behavior'):
getattr(a, name)
np_config.enable_numpy_behavior()
for name in {
'T', 'astype', 'ravel', 'transpose', 'reshape', 'clip', 'size',
'tolist'
}:
_ = getattr(a, name)
from tensorflow.python.ops.numpy_ops import *
from tensorflow import bfloat16
except ImportError:
# Note that this fallback will be needed for TF 2.2.
from tensorflow import newaxis
from trax.tf_numpy.numpy_impl import random
# pylint: disable=wildcard-import
from trax.tf_numpy.numpy_impl.array_ops import *
from trax.tf_numpy.numpy_impl.arrays import *
from trax.tf_numpy.numpy_impl.dtypes import *
from trax.tf_numpy.numpy_impl.math_ops import *
from trax.tf_numpy.numpy_impl.utils import finfo
from trax.tf_numpy.numpy_impl.utils import promote_types
from trax.tf_numpy.numpy_impl.utils import result_type
# pylint: enable=wildcard-import
max = amax # pylint: disable=redefined-builtin,undefined-variable
min = amin # pylint: disable=redefined-builtin,undefined-variable
round = around # pylint: disable=redefined-builtin,undefined-variable
try:
from tensorflow.python.ops.numpy_ops.np_config import enable_numpy_behavior
# TODO(b/171429739): This should be moved to every individual file/test.
enable_numpy_behavior()
except ImportError:
pass
from tensorflow.keras.callbacks import ModelCheckpoint, TensorBoard
from tensorflow.keras import regularizers, activations, initializers, constraints, Sequential
from tensorflow.keras import backend as K
from tensorflow.keras.constraints import UnitNorm, Constraint
import tensorflow.experimental.numpy as tnp
# from AE import Sampling
# from AE import VAE
from tensorflow import keras
from tensorflow.keras import layers
from tensorflow.python.ops.numpy_ops import np_config
np_config.enable_numpy_behavior()
# Could be useful to implement talos library for gidsearch to run on the weekend
class DenseTied(Layer):
def __init__(self, units,
activation=None,
use_bias=True,
kernel_initializer='glorot_uniform',
bias_initializer='zeros',
kernel_regularizer=None,
bias_regularizer=None,
activity_regularizer=None,
kernel_constraint=None,
bias_constraint=None,
