美文网首页
tensorflow

tensorflow

作者: 王超_vip | 来源:发表于2019-04-09 16:22 被阅读0次

    1. reduce_sum应该理解为压缩求和,用于降维

    # 'x' is [[1, 1, 1]

    #         [1, 1, 1]]

    #求和

    tf.reduce_sum(x) ==> 6

    #按列求和

    tf.reduce_sum(x, 0) ==> [2, 2, 2]

    #按行求和

    tf.reduce_sum(x, 1) ==> [3, 3]

    #按照行的维度求和

    tf.reduce_sum(x, 1, keep_dims=True) ==> [[3], [3]]

    #行列求和

    tf.reduce_sum(x, [0, 1]) ==> 6

    ---------------------

    2.tf.argmax()解析

    axis = 0:

    axis=0时比较每一列的元素,将每一列最大元素所在的索引记录下来,最后输出每一列最大元素所在的索引数组。

    axis = 1:

    axis=1的时候,将每一行最大元素所在的索引记录下来,最后返回每一行最大元素所在的索引数组。

    3. tf.cast()数据类型转换

    cast(x, dtype, name=None)

    第一个参数 x:   待转换的数据(张量)

    第二个参数 dtype: 目标数据类型

    第三个参数 name: 可选参数,定义操作的名称

    相关文章

      网友评论

          本文标题:tensorflow

          本文链接:https://www.haomeiwen.com/subject/rghiiqtx.html