如何生成N张随机张量堆栈?

时间:2019-12-06 08:07:59

标签: tensorflow

pythonic的方式就是写:

[tf.random.uniform([3,3]) for i in range(5)]

在我的情况下,tf.random.uniform([3,3])将更改为myFunction(),它会返回恒定形状的随机张量。

我可以做:

myStack = tf.stack([myFunction() for i in range(5)])

但这表现很糟。

将这些5 x [3,3]张量堆叠起来是一种优雅的张量流方式吗? (返回值应该是[3,3]张量的集合。由于性能不佳,显然它不能是python列表)

我可以使用tf.tiletf.while_loop使它起作用,但这感觉像是太多的黑客攻击。 有没有干净的方法?

0 个答案:

没有答案
相关问题