Python – tensorflow.DeviceSpec.replace()

TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning  neural networks.

replace() is used to override the specification of DeviceSpec object and get the new object.

Syntax: tensorflow.DeviceSpec.replace(**kwargs)

Parameters:

• **kwargs: This method accepts all the parameters accepted by DeviceSpec constructor.

Returns: It returns a DeviceSpec object.

Example 1:

Python3

# Importing the library
import tensorflow as tf

# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg", replica = 5)

# Printing the DeviceSpec
print('Device Spec: ', device_spec.to_string())

# Getting new DeviceSpec object
new_device_spec = device_spec.replace(job = "gfg2")

# Printing the result
print('New Device Spec: ', new_device_spec.to_string())

Output:

Device Spec:  /job:gfg/replica:5
New Device Spec:  /job:gfg2/replica:5

Example 2:

Python3

# Importing the library
import tensorflow as tf

# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg", replica = 5)

# Printing the DeviceSpec
print('Device Spec: ', device_spec.to_string())

# Getting new DeviceSpec object
new_device_spec = device_spec.replace(job = "gfg2", device_type = "CPU")

# Printing the result
print('New Device Spec: ', new_device_spec.to_string())

Output:

Device Spec:  /job:gfg/replica:5
New Device Spec:  /job:gfg2/replica:5/device:CPU:*