Python – tensorflow.IndexedSlices.device Attribute

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

device is used to find the name of the device on which values will be generated.

Syntax: tensorflow.IndexedSlices.device

Returns: It returns the name of the device.

Example 1:

Python3

# Importing the library
import tensorflow as tf

# Initializing the input
data = tf.constant([1, 2, 3])

# Printing the input
print('data: ', data)

# Calculating result
res = tf.IndexedSlices(data, [0])

# Finding device name
device = res.device

# Printing the result
print('device: ', device)

Output:

data:  tf.Tensor([1 2 3], shape=(3, ), dtype=int32)
device:  /job:localhost/replica:0/task:0/device:CPU:0

Example 2:

Python3

# Importing the library
import tensorflow as tf

# Initializing the input
data = tf.constant([[1, 2, 3], [4, 5, 6]])

# Printing the input
print('data: ', data)

# Calculating result
res = tf.IndexedSlices(data, [0])

# Finding device name
device = res.device

# Printing the result
print('device: ', device)

Output:

data:  tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32)
device:  /job:localhost/replica:0/task:0/device:CPU:0