Align query_chw and get_dimensions

[pmeier]

We have to proper combination

vision/torchvision/transforms/v2/functional/_meta.py

Line 84 in 9b82df4

def get_size(inpt: datapoints._InputTypeJIT) -> List[int]:

vision/torchvision/transforms/v2/utils.py

Line 35 in 9b82df4

def query_size(flat_inputs: List[Any]) -> Tuple[int, int]:

However, we have the combination below that diverges on the name

vision/torchvision/transforms/v2/functional/_meta.py

Line 15 in 9b82df4

def get_dimensions(inpt: Union[datapoints._ImageTypeJIT, datapoints._VideoTypeJIT]) -> List[int]:

vision/torchvision/transforms/v2/utils.py

Line 21 in 9b82df4

def query_chw(flat_inputs: List[Any]) -> Tuple[int, int, int]:

I don't recall anymore how we ended up with this, but we should correct it.

My vote is out for also calling it query_dimensions since we already have

vision/torchvision/transforms/functional.py

Line 64 in 9b82df4

def get_dimensions(img: Tensor) -> List[int]:

in v1.

cc @vfdev-5

[Brooklynn29]

Hello !, I'm looking forward to contribute, how can I do that ?

[NicolasHug]

Can we get rid of the query* stuff and just let get* accept both single datapoints/tensors and flat inputs?

And BTW, since we're not providing public APIs to flatten / unflatten inputs nor to handle arbitrary structures in custom transforms, I don't think we need to make query* public (should we keep it).