2025年9月25日
Torch中的tensor size
摘要： Torch中的tensor size 本文重点介绍了一下如何在PyTorch中去计算一个高维tensor的大小，也就是元素的总数。在其他框架中我们需要使用size函数来获取，而在PyTorch框架中这个接口被调整为numel，本文给出了两个具体代码示例。 阅读全文
posted @ 2025-09-25 12:31 DECHIN 阅读(20) 评论(0) 推荐(0)