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)