比方说,a是一个已经定义好的tensor张量
那么直接:
a.numpy()即可。

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐