Is there a way to visualize the network structure with TensorBoard like the graph in TorchStudio?

Hi, guys,
I am wondering whether there is a way to visualize the network structure with TensorBoard like the graph in TorchStudio.
And here is a screenshot of the graph from TorchStudio,

Your answer and guide will be appreciated!