torch_complex_tensor_initialization_and_basic_arithmetic_operations.py

1import torch
2from torch_complex.tensor import ComplexTensor
3
4# Create real and imaginary parts as standard PyTorch tensors
5real = torch.ones(3, 3)
6imag = torch.zeros(3, 3)
7
8# Initialize a ComplexTensor
9z = ComplexTensor(real, imag)
10
11# Perform basic operations
12# Addition
13z_sum = z + z
14
15# Multiplication
16z_prod = z * 2.0
17
18# Matrix multiplication
19z_matmul = z.matmul(z)
20
21print("Complex Tensor:")
22print(z)
23print("Real part:", z.real)
24print("Imaginary part:", z.imag)