【问题标题】:Reduce the rank of a one one tensor of Tensorflow?降低Tensorflow的一对一张量的等级?
【发布时间】:2018-11-14 04:33:52
【问题描述】:

我有一个 Tensorflow 张量 [[1]],如何将这个张量更改为 [1],它只是一个数字?

非常感谢!

【问题讨论】:

  • 你能和我们分享一下你到目前为止的尝试吗?
  • [1] 不是一个数字,它是一个包含一个元素的列表(或一维张量)。
  • 我需要在我的代码中执行 tf.matmul(A,wt1)) ,其中 A 类似于 [[1,2,3]] 和 wt = tf.Variable(tf.truncated_normal( [filter_size, 1],平均值 = 0.0,stddev = 1.0,dtype = tf.float64))。我想存储 matmul 的输出(类似于 [[1]]),但我不知道如何存储。谢谢!

标签: python tensorflow


【解决方案1】:

在这里,我创建了一个虚拟 2D 张量并将其转换为您要查找的标量。请注意,您可以在会话中运行它,tf.Session().run(tensor) 而不是下面的 InteractiveSession

这里要注意的一点是,Session.runeval 的任何返回张量都是一个 numpy 数组。如果返回的数组是多维的,您可以调用np.ravel() 将其设为一维,并将第一个元素设为标量。

>>> sess = tf.InteractiveSession()
>>> t=[[10]]
>>> tensor=tf.convert_to_tensor(t)
>>> tensor.eval().ravel()[0]
10

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2016-06-20
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多