TensorFlow
关注
分享
扫一扫
正门大石狮
写可以复现的实验，写可以看懂的理论。
展开
-
【BUG】Keras用了multi_gpu_model，但用Nvidia-smi看只用了一块显卡GPU
解决方法很简单，在你原来使用multi_gpu_model外层加上两句代码：strategy = tf.distribute.MirroredStrategy(devices=["/gpu:0","/gpu:1", "/gpu:2"])with strategy.scope():原创 2020-11-25 12:36:17 · 1732 阅读 · 8 评论 -
TensorFlow中的tf.slice()是个什么操作
tf.slice() 定义如下tf.slice( input_, begin, size, name=None)参数意义input_一个Tensorbegin各个维度的起始坐标组成的向量，从0开始size各个维度取出来的值的个数所组成的向量，从1开始示例一个Tensor如下所示：其每个维度上值的个数分别是3、2、3个（由外到内）。t ...原创 2020-04-28 15:53:40 · 524 阅读 · 0 评论