experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use arg 'axis' instead of deprecated arg 'dim'

Browse Source
This commit is contained in:
Robert Herbig 2019-09-16 22:35:50 -04:00
parent 376e283836
commit bb69697254
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/ops/clustering_ops.py
View File

@ -216,11 +216,11 @@ class KMeans(object):
output = [] output = []
if not inputs_normalized: if not inputs_normalized:
with ops.colocate_with(clusters, ignore_existing=True): with ops.colocate_with(clusters, ignore_existing=True):
clusters = nn_impl.l2_normalize(clusters, dim=1) clusters = nn_impl.l2_normalize(clusters, axis=1)
for inp in inputs: for inp in inputs:
with ops.colocate_with(inp, ignore_existing=True): with ops.colocate_with(inp, ignore_existing=True):
if not inputs_normalized: if not inputs_normalized:
inp = nn_impl.l2_normalize(inp, dim=1) inp = nn_impl.l2_normalize(inp, axis=1)
output.append(1 - math_ops.matmul(inp, clusters, transpose_b=True)) output.append(1 - math_ops.matmul(inp, clusters, transpose_b=True))
return output return output
@ -251,7 +251,7 @@ class KMeans(object):
# TODO(ands): Support COSINE distance in nearest_neighbors and remove # TODO(ands): Support COSINE distance in nearest_neighbors and remove
# this. # this.
with ops.colocate_with(clusters, ignore_existing=True): with ops.colocate_with(clusters, ignore_existing=True):
clusters = nn_impl.l2_normalize(clusters, dim=1) clusters = nn_impl.l2_normalize(clusters, axis=1)
for inp, score in zip(inputs, scores): for inp, score in zip(inputs, scores):
with ops.colocate_with(inp, ignore_existing=True): with ops.colocate_with(inp, ignore_existing=True):
(indices, distances) = gen_clustering_ops.nearest_neighbors( (indices, distances) = gen_clustering_ops.nearest_neighbors(