reduction_indices in tensorflow

https://www.cnblogs.com/likethanlove/p/6547405.html