使用tf.tile或tf.expand_dims以某种方式复制张量

Question

基于tensorflow函数。 张量A = [1 10 5 6]。我如何创建张量B，如下所示：

[[[1  1  1 ],[1  1  1 ],[1  1  1 ]],
 [[10 10 10],[10 10 10],[10 10 10]],
 [[5  5  5 ],[5  5  5 ],[5  5  5 ]],
 [[6  6  6 ],[6  6  6 ],[6  6  6 ]]]

谢谢。

Answer 1

A_reshape = tf.expand_dims(A,1)
A_reshape = tf.expand_dims(A_reshape,2)
B = tf.tile(A_reshape, [1,3,3])