Proper drawing of gradients

Gradient of any operation w.r.t. TensorFST is a NamedTuple, we need to convert it back to TensorFST in order to draw it nicely.