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.tile
,tf.while_loop
使它起作用,但这感觉像是太多的黑客攻击。
有没有干净的方法?