求教Pytorch中使用循环更快还是扩大矩阵大小处理更快?

是这样 我现在需要使用pytorch自己写一个module

我是采用_for_循环将tensor循环处理很多遍比较快还是直接_ repeat _扩大tensor再处理比较快?

谢谢各位大佬!!(抱歉一直在csdn潜水所以没有悬赏可以提供)

循环肯定慢,因为循环不能向量化,所以是串行计算的。你对比测试下就知道。

越高级的语言执行for 循环就越慢,torch是使用c语言编写的,底层也是for循环,但肯定比Python语言执行更快