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

Question

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

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

谢谢各位大佬！！（抱歉一直在csdn潜水所以没有悬赏可以提供）

Answer 1

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