Tensorflow Tensor的属性

Tensorflow Tensor的属性

名称	说明
device	设备位置
numpy	numpy数据
shape	数据形状
ndim	数据维度
rank	数据维度(返回tensor类型)
name	名称（tf2.0无效了）

# device 设备位置
with tf.device("cpu"):
    a = tf.constant([1])

print(a.device)

b= tf.range(4)
# numpy  直接获取 Tensor中的 numpy数据
print(a.numpy())
print(b.numpy())

# ndim  数据的维度
print(a.ndim)
print(b.ndim)

# shape 数据的形状
print(a.shape)
print(b.shape)

# rank  数据的维度(tensor)格式
print(tf.rank(a))
print(tf.rank(b))

# name 已经无效
print(a.name)