如何在TensorFlow中提取张量纯数值（排除类型与形状信息）

在Google Colaboratory里训练完回归模型后，要把w、b这类张量里的纯数值提取出来，有几种简单直接的方式：

• 用.numpy()方法（通用场景）
  这是最常用的方式，直接对标量张量调用该方法就能拿到对应的Python数值，示例代码如下：

  import tensorflow as tf
  
  # 假设训练得到的w、b是标量张量
  w = tf.constant(3.2996964)
  b = tf.constant(99.626526)
  
  slope = w.numpy()
  intercept = b.numpy()
  
  print(f"Slope = {slope}, Intercept = {intercept}")
  

• 用.item()方法（单值张量专属）
  如果你的张量是单元素张量（比如形状为(1,)的张量），用.item()能更直接地提取纯数值，不用额外处理形状：

  # 假设w是形状为(1,)的张量
  w = tf.constant([3.2996964])
  slope = w.item()
  
  # 标量张量也可以用这个方法
  b = tf.constant(99.626526)
  intercept = b.item()
  
  print(f"Slope = {slope}, Intercept = {intercept}")
  

如果你的张量在GPU上，.numpy()和.item()都会自动把数据转到CPU，不用额外做设备转换；要是张量是多元素的，.numpy()会返回对应的NumPy数组，按需索引提取元素就行。

内容的提问来源于stack exchange，提问作者zo333