with tf.Session() as sess: with tf.device("/gpu:0"): .................
with tf.Session() as sess:
with tf.device("/gpu:0"):
.................
a是一个tensor，如果要输出a的值，则print(a.eval())
本文标题：tf
本文链接：https://www.haomeiwen.com/subject/vildtqtx.html
网友评论