ttnn.complex_tensor

ttnn.complex_tensor() → None

Create a complex tensor from real and imaginary part tensors.

:param * real: :param * imag:

Example

>>> real = ttnn.to_device(ttnn.from_torch(torch.tensor((1, 2), dtype=torch.bfloat16)), device)
>>> imag = ttnn.to_device(ttnn.from_torch(torch.tensor((1, 2), dtype=torch.bfloat16)), device)
>>> complex_tensor = ttnn.complex_tensor(real, imag)