深度学习API
飘散风中
这个作者很懒,什么都没留下…
展开
-
【深度学习API】TensorFlow - tf.concat()
tf.concat()作用:拼接张量,主要参数为values,axis;values为需要合并的listaxis决定了这些list中的张量如何合并,axis=0,合并第一维,axis=1,合并第二维,axis=2合并第三维,axis=3合并第四维度,前提是,除了需要合并的那一位纬度,其他不合并的纬度属性必须相同;axis=3时,四个张量最后一维不同,合并第四维,代码: ...原创 2019-01-25 19:26:31 · 858 阅读 · 0 评论 -
【深度学习API】TensorFlow - tf.argmax()
tf.argmax()作用:寻找张量中的最大值,并返回起下标,主要参数为:input, axisinput:输入的张量axis:决定了在第几纬度找最大值,例如,axis = 0,决定了第一纬度找最大值,axis = 1 ,第二维度找最大值(此时第一维度的大小不变),axis 不可以超过张量的最大维度(shape = (1,5),那么axis 选择只有0和1,shape为(1,2,5),...原创 2019-01-28 18:36:07 · 327 阅读 · 0 评论 -
【深度学习API】TensorFlow - tf.one_hot()
tf.one_hot()作用:转换为one-hot 编码格式,由于我们一般预测结尾使用softmax,导致结果全为one-hot形式,因此我们在做测试集时,需要将label转换为one-hot格式,或者将预测结果的one-hot格式转换为数组形式;关键参数:indices , depthindices: 传入tensor,如[1,0,3,2]depth:one-hot的编码深度...原创 2019-01-29 16:10:22 · 394 阅读 · 0 评论