Thanks, now I understand when/how the dims are used - will stick to using positional indexing instead inside of the model.