前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >tf.convert_to_tensor()

tf.convert_to_tensor()

作者头像
狼啸风云
修改2022-09-04 21:46:48
4.1K0
修改2022-09-04 21:46:48
举报

将给定值转换为张量。

代码语言:javascript
复制
tf.convert_to_tensor(
    value,
    dtype=None,
    name=None,
    preferred_dtype=None,
    dtype_hint=None
)

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

代码语言:javascript
复制
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列表和标量。

注意:当Python列表或标量中不存在浮点数和字符串类型时,此函数与默认Numpy行为不同。将抛出一个错误,而不是静静地转换None值。

参数:

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

返回值:

  • 一个基于值的张量。

可能产生的异常:

  • 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
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019年07月29日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档