code: def channel_view(x:Tensor)->Tensor:
      "Make channel the first axis of `x` and flatten remaining axes"
      return x.transpose(0,1).contiguous().view(x.shape[1],-1f channel_view(x:Tensor)->Tensor:
      "Make channel the first axis of `x` and flatten remaining axes"
      return x.transpose(0,1).contiguous().view(x.shape[1],-1)
docs: Make channel the first axis of `x` and flatten remaining axes