Hello Sionna

Hello Sionna!

In this post, basic implemenation of the Sionna library is discussed.

import os
if os.getenv("CUDA_VISIBLE_DEVICES") is None:
    gpu_num = 0 # Use "" to use the CPU
    os.environ["CUDA_VISIBLE_DEVICES"] = f"{gpu_num}"
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

try:
    import sionna
except ImportError as e:
    # Install Sionna if package is not already installed
    import os
    os.system("pip install sionna")
    import sionna

# IPython "magic function" for inline plots
%matplotlib inline
import matplotlib.pyplot as plt

batch_size = 1000 # Number of symbols we want to generate
num_bits_per_symbol = 4 # 16-QAM has four bits per symbol
binary_source = sionna.utils.BinarySource()
b = binary_source([batch_size, num_bits_per_symbol])
b

<tf.Tensor: shape=(1000, 4), dtype=float32, numpy=
array([[1., 0., 0., 1.],
       [0., 1., 0., 1.],
       [1., 1., 1., 0.],
       ...,
       [1., 0., 1., 1.],
       [0., 0., 1., 1.],
       [1., 0., 0., 1.]], dtype=float32)>

constellation = sionna.mapping.Constellation("qam", num_bits_per_symbol)
constellation.show();

mapper = sionna.mapping.Mapper(constellation=constellation)
x = mapper(b)
x[:10]

<tf.Tensor: shape=(10, 1), dtype=complex64, numpy=
array([[-0.3162278+0.9486833j],
       [ 0.3162278-0.9486833j],
       [-0.9486833-0.3162278j],
       [-0.9486833+0.9486833j],
       [ 0.3162278+0.3162278j],
       [-0.3162278-0.3162278j],
       [-0.3162278+0.9486833j],
       [ 0.3162278+0.9486833j],
       [ 0.9486833-0.3162278j],
       [ 0.3162278-0.3162278j]], dtype=complex64)>

awgn = sionna.channel.AWGN()
ebno_db = 15 # Desired Eb/No in dB
no = sionna.utils.ebnodb2no(ebno_db, num_bits_per_symbol, coderate=1)
y = awgn([x, no])

# Visualize the received signal
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure(figsize=(7,7))
ax = fig.add_subplot(111)
plt.scatter(np.real(y), np.imag(y));
ax.set_aspect("equal", adjustable="box")
plt.xlabel("Real Part")
plt.ylabel("Imaginary Part")
plt.grid(True, which="both", axis="both")
plt.title("Received Symbols");