张量图不显示张量形状

Question

我正在使用python创建一个简单的模型，我想可视化模型的图形。我正在使用张量板显示图形，但是它没有在图形的各层中显示张量形状。有没有办法显示形状？ 谢谢。

这是我尝试过的简单代码（不是扩展模型，但我敢肯定，我可以从此代码的解决方案中概括出模型的解决方案）。

import React, { useCallback } from 'react';

const ExampleComponent = (props) => {
  const onClick = useCallback(() => {
    console.log('got clicked', props.name);
  }, [props.name]);

  return (
    <SomeOtherComponent onClick={onClick} />
  )
}