(x)
return torch.unsqueeze(torch.transpose(x, 1, 2), 3)
Tranform the size of the tensors to fit for conv input.
Tranform the size of the tensors to fit for conv input.
Tranform the size of the tensors to fit for conv input .
def shape_transform(x):
    """ Tranform the size of the tensors to fit for conv input. """
    return torch.unsqueeze(torch.transpose(x, 1, 2), 3)
def shape_transform ( x ) : return torch . unsqueeze ( torch . transpose ( x , 1 , 2 ) , 3 )