meddlr.ops.complex.channels_last

meddlr.ops.complex.channels_last(x: Tensor)

Permute complex-valued x to channels-last convention.

Parameters

x (torch.Tensor) – A tensor of shape [B,C,H,W,…] or [B,C,H,W,…,2].

Returns

A tensor of shape [B,H,W,…,C] or [B,H,W,…,C,2].

Return type

torch.Tensor