tf

with tf.Session() as sess:

    with tf.device("/gpu:0"):

        .................

a是一个tensor,如果要输出a的值,则print(a.eval())

推荐阅读更多精彩内容