keras:用无尺寸重整张量

时间:2019-02-28 10:49:48

标签: python tensorflow keras tensor

假设我有一个(None, a, b*c)形状的张量,并且我想对其进行重整以得到(None, a, b, c)的形状。我知道我能做到吗?

1 个答案:

答案 0 :(得分:0)

假设张量名为t并且形状为(None, a, b*c),则可以执行以下操作:

 x_reshaped = tf.reshape(t, [-1, a, b, c])