当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tensorflow.DeviceSpec.make_merged_spec()用法及代码示例


TensorFlow是Google设计的开源Python库,用于开发机器学习模型和深度学习神经网络。

make_merged_spec用于合并两个DeviceSpec的规范。

用法:tensorflow.DeviceSpec.make_merged_spec( dev )

参数:

  • dev:它是DeviceSpec。

返回值:它返回具有合并规范的DeviceSpec对象。



范例1:

Python3

# Importing the library 
import tensorflow as tf 
  
# Initializing Device Specification 
device_spec = tf.DeviceSpec(job ="gfg", replica = 5) 
device_spec2 = tf.DeviceSpec(device_type ="CPU") 
  
# Printing the DeviceSpec object 
print('Device Spec:', device_spec) 
print('Device Spec2:', device_spec2) 
  
# Getting new DeviceSpec object 
new_device_spec = device_spec.make_merged_spec(device_spec2) 
  
# Printing the result 
print('New Device Spec:', new_device_spec)

输出:


Device Spec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fad779e4d08>
Device Spec2: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fadb9428228>
New Device Spec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fad779e4ca8>

范例2:

Python3

# Importing the library 
import tensorflow as tf 
  
# Initializing Device Specification 
device_spec = tf.DeviceSpec(job ="gfg", replica = 5) 
device_spec2 = tf.DeviceSpec(device_type ="CPU", device_index = 2) 
  
# Printing the DeviceSpec object 
print('Device Spec:', device_spec) 
print('Device Spec2:', device_spec2) 
  
# Getting new DeviceSpec object 
new_device_spec = device_spec.make_merged_spec(device_spec2) 
  
# Printing the result 
print('New Device Spec:', new_device_spec)

输出:


Device Spec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fad779e4e28>
Device Spec2: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fad779e4d08>
New Device Spec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fadb9428228>




相关用法


注:本文由纯净天空筛选整理自aman neekhara大神的英文原创作品 Python – tensorflow.DeviceSpec.make_merged_spec()。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。