tensorflow python中的变形字典

nuypyhwy  于 2022-11-16  发布在  Python
关注(0)|答案(1)|浏览(128)

我有一个字典,有两个键,每个键的形状是([612,41,41,1])。我想把它重新整形,使其形状为([612,41,41,2]),以便在cnn中输入它,输入通道= 2。如果我使用命令tf.expand_dims(my_dict[1],-2)}不起作用。有什么想法吗?

rta7y2nd

rta7y2nd1#

在我看来,因为612x41x41x1不同于612x41x41x2,所以你不能重塑到这个形状。但如果你的意思是你想复制最后一个维度,所以你可以尝试以下代码:

my_tensor = tf.tile(my_tensor, (1, 1, 1, 2))

相关问题