tf.device

tf.device(device_name)

device_name:要在上下文中使用的设备名称。

with tf.device(device_name):

指定使用设备。 

猜你喜欢

转载自blog.csdn.net/Fwuyi/article/details/125660997
tf