Update 03_tensor.py