qq_35148758的博客

tensorflow常用函数整理1 tf.reshape2 tf.concat()3.tf.expand_dims记录自己使用过的tensorflow常用函数，以便以后查询1 tf.reshapetf.reshape( tensor, shape, name=None)将给定tensor的维度转换为shape。arr = tf.Variable([[1,2],...

tensorflow在训练时默认占用所有GPU的显存。可以通过以下方式解决该问题：1.在构造tf.Session()时候通过传递tf.GPUOptions作为可选配置参数的一部分来显式地指定需要分配的显存比例，如下所示：# 假如有12GB的显存并使用其中的4GB:gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.33...