diff options
author | 2018-04-17 01:00:44 +0800 | |
---|---|---|
committer | 2018-04-17 01:00:44 +0800 | |
commit | c22d996c3d6a16db292bd3464b2ef7b91adae676 (patch) | |
tree | 131b65fe464f9f5bf2724117ca6f90f38680f859 /tensorflow/contrib/timeseries | |
parent | 63c6562df68ade3a03481874a71b536a4e02b6f5 (diff) |
Fix expand_dims of dims argument has been deprecated with axis
Diffstat (limited to 'tensorflow/contrib/timeseries')
-rw-r--r-- | tensorflow/contrib/timeseries/python/timeseries/state_space_models/state_space_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/timeseries/python/timeseries/state_space_models/state_space_model.py b/tensorflow/contrib/timeseries/python/timeseries/state_space_models/state_space_model.py index 951c6546d5..d04c721007 100644 --- a/tensorflow/contrib/timeseries/python/timeseries/state_space_models/state_space_model.py +++ b/tensorflow/contrib/timeseries/python/timeseries/state_space_models/state_space_model.py @@ -909,7 +909,7 @@ class StateSpaceModel(model.SequentialTimeSeriesModel): elif unbroadcasted_shape.ndims == 2: # Unbroadcasted shape [num features x state dimension] broadcasted_model = array_ops.tile( - array_ops.expand_dims(unbroadcasted_model, dim=0), + array_ops.expand_dims(unbroadcasted_model, axis=0), [array_ops.shape(times)[0], 1, 1]) elif unbroadcasted_shape.ndims == 3: broadcasted_model = unbroadcasted_model |