0

The rank is just not the same: [?, 28, 28, 1] and [?, 10]. How should I deal with this

Werma
  • 53
  • 1
  • 9

1 Answers1

0

Just reshape your [?, 28, 28, 1] tensor into [?, 28*28*1] and concatenate it with [?, 10] at axis=1, so you'll have a new tensor of shape [?, (28*28*1 + 10)] then you can use CVAE as usual.

Thaillie
  • 1,362
  • 3
  • 17
  • 31
古定華
  • 1
  • 1