Tensorflow学习笔记参考—源码分析之最近算法

[python] view plaincopy

reference: tensorflow专栏系列待看

http://blog.csdn.net/column/details/tensorflow.html

源代码地址:https://github.com/aymericdamien/TensorFlow-Examples/blob/master/examples/2%20-%20Basic%20Classifiers/nearest_neighbor.py

相关API&＃xff1a;

tf.reduce_sum(input_tensor, reduction_indices&＃61;None, keep_dims&＃61;False, name&＃61;None)

Computes the sum of elements across dimensions of a tensor.

Reduces input_tensor along the dimensions given in reduction_indices. Unless keep_dims is true, the rank of the tensor is reduced by 1 for each entry in reduction_indices. If keep_dims is true, the reduced dimensions are retained with length 1.

If reduction_indices has no entries, all dimensions are reduced, and a tensor with a single element is returned.

For example:

# &＃39;x&＃39; is [[1, 1, 1]
# [1, 1, 1]]
tf.reduce_sum(x) &＃61;&＃61;> 6
tf.reduce_sum(x, 0) &＃61;&＃61;> [2, 2, 2]
tf.reduce_sum(x, 1) &＃61;&＃61;> [3, 3]
tf.reduce_sum(x, 1, keep_dims&＃61;True) &＃61;&＃61;> [[3], [3]]
tf.reduce_sum(x, [0, 1]) &＃61;&＃61;> 6


Args:

• input_tensor: The tensor to reduce. Should have numeric type.
• reduction_indices: The dimensions to reduce. If None (the default), reduces all dimensions.
• keep_dims: If true, retains reduced dimensions with length 1.
• name: A name for the operation (optional).

Returns:

The reduced tensor.