What's the position mean in TransUNet3D?

[yjsb]

I just want to use your 3D TransUNet in your code.
but I don't know the input shape of the position parameter in forward, I use
position=torch.Tensor([1,512,2,2,2, 64])
to match the
self.pos_emb_layer = nn.Linear(6, emb_channels)
I feel I was wrong.