TensorFlow - Tensor Reshape

Question

我正在使用TensorFlow，我有以下矩阵：

U2 = tf.constant([[1,3],[0,1],[1,0]],dtype=tf.float32)

[[ 1.  3.]
 [ 0.  1.]
 [ 1.  0.]]

重塑此矩阵的最简单方法是什么，以便产生张量：

tf.constant([[[1,1],[0,0],[1,1]],[[3,3],[1,1],[0,0]]],dtype=tf.float32)

[[[ 1.  1.]
  [ 0.  0.]
  [ 1.  1.]]

 [[ 3.  3.]
  [ 1.  1.]
  [ 0.  0.]]]

Answer 1

这是使用TensorFlow API创建第一个子矩阵的一种快捷方法：

U2 = tf.constant([[1,3],[0,1],[1,0]],dtype=tf.float32)
first_col = tf.unpack(U2, axis=1)[0]
repeated_cols = tf.tile(first_col, [2])
repeated_cols = tf.reshape(repeated_cols, [3,2])

哪个会创建

[[ 1.  1.]
  [ 0.  0.]
  [ 1.  1.]]