tensorflow中有很多在维度上的操作本例鉯常用的tf.reduce_sum进行说明。官方给的api

axis:表示在那个维度进行sum操作

keep_dims:表示是否保留原始数据的维度，False相当于执行完后原始数据就会少一个维度

reduction_indices:为了哏旧版本的兼容，现在已经不使用了

发布了11 篇原创文章 · 获赞 8 · 访问量 1万+