pytorch取tensor的对角线上的元素

sullay

关注

阅读 115

2023-03-17


使用 ​​diag()​​函数。示例如下:

pytorch取tensor的对角线上的元素_人工智能


相关推荐

精彩评论(0)

0 0 举报