Searched refs:reshape_tensor_complex (Results 1 – 1 of 1) sorted by relevance
410 def reshape_tensor_complex(tensor: torch.Tensor) -> torch.Tensor: function427 x_reshaped = reshape_tensor_complex(x.view(x.real.dtype))428 z_reshaped = reshape_tensor_complex(z.view(y.real.dtype))