TensorFlow将给定值转换为张量
tf.convert_to_tensor
此函数将各种类型的 Python 对象转换为 Tensor 对象.它接受 Tensor 对象,numpy 数组,Python 列表和 Python 标量
convert_to_tensor (
value ,
dtype=None ,
name=None ,
preferred_dtype=None
)
输入格式类型
importnumpy as npdefmy_func(arg):
arg= tf.convert_to_tensor(arg, dtype=tf.float32)return tf.matmul(arg, arg) +arg#The following calls are equivalent.
value_1 = my_func(tf.constant([[1.0, 2.0], [3.0, 4.0]]))
value_2= my_func([[1.0, 2.0], [3.0, 4.0]])
value_3= my_func(np.array([[1.0, 2.0],