1. 程式人生 > >tensorflow中的.numpy()方法

tensorflow中的.numpy()方法

.numpy方法只用在使用tf.enable_eager_execution()(指令式程式設計開啟)後才有的方法, 否則會有==AttributeError: ‘Tensor’ object has no attribute ‘numpy’==報錯

from math import pi

def f(x):
  return tf.square(tf.sin(x))

assert f(pi/2).numpy() == 1.0
print(f(pi/2).numpy())
print(f(pi/2))

輸出:

1.0
tf.Tensor(1.0, shape=(), dtype=float32)