I’m want to create a network which outputs a tensor shaped like this:
batch x X x 6: where X is a number which is between 1 and 50
Sort of like a network which predicts the bounding boxes for a give shape in an image. But Im not sure which layer should be the last, because for example a nn.Linear() have a fixed output size.
Thank you!
DC