tf.convert_to_tensor

参考  tf.convert_to_tensor - 云+社区 - 腾讯云

将给定值转换为张量。

tf.convert_to_tensor(
    value,
    dtype=None,
    dtype_hint=None,
    name=None
)

该函数将各种类型的Python对象转换为张量对象。它接受张量对象、数字数组、Python列表和Python标量。

例：

import numpy as np

def my_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], [3.0, 4.0]], dtype=np.float32))

这个函数在用Python编写新操作时非常有用(如上面示例中的my_func)。所有标准的Python op构造函数都将此函数应用于它们的每个张量值输入，这使得这些ops除了接受张量对象外，还可以接受numpy数组、Python列表和标量。

参数：

• value:类型具有注册张量转换函数的对象。
• dtype:返回张量的可选元素类型。如果缺少，则从值的类型推断类型。
• dtype_hint:返回张量的可选元素类型，当dtype为None时使用。在某些情况下，调用者在转换为张量时可能没有考虑到dtype，因此dtype_hint可以用作软首选项。如果不能转换为dtype_hint，则此参数没有效果。
• name:创建新张量时使用的可选名称。

返回值：

• 一个基于值的张量。

可能产生的异常：

• TypeError: If no conversion function is registered for value to dtype.
• RuntimeError: If a registered conversion function returns an invalid value.
• ValueError: If the value is a tensor not of given dtype in graph mode.