Functionnchw_to_nlcFlatten [N, C, H, W] shape tensor to [N, L, C] shape tensor. Args: x (Tensor): The input tensor of shape [N, C, H, W] before convertion.
mmseg/models/utils/shape_convert.py:21
Functionnlc_to_nchwConvert [N, L, C] shape tensor to [N, C, H, W] shape tensor. Args: x (Tensor): The input tensor of shape [N, L, C] before convertion.
mmseg/models/utils/shape_convert.py:4