我有一个字典,有两个键,每个键的形状是([612,41,41,1])。我想把它重新整形,使其形状为([612,41,41,2]),以便在cnn中输入它,输入通道= 2。如果我使用命令tf.expand_dims(my_dict[1],-2)}不起作用。有什么想法吗?
rta7y2nd1#
在我看来,因为612x41x41x1不同于612x41x41x2,所以你不能重塑到这个形状。但如果你的意思是你想复制最后一个维度,所以你可以尝试以下代码:
my_tensor = tf.tile(my_tensor, (1, 1, 1, 2))
1条答案
按热度按时间rta7y2nd1#
在我看来,因为612x41x41x1不同于612x41x41x2,所以你不能重塑到这个形状。但如果你的意思是你想复制最后一个维度,所以你可以尝试以下代码: