Pytorch简单
文章平均质量分 50
音程
这个作者很懒,什么都没留下…
展开
-
如何实现torch.arange的tensor版本
因为这样才可以保证,输出的结果的维度是一样的。步长一个是1,一个是2,这样肯定不行,两个。输出的维度不同,就肯定不可能并行。细心的人可以发现,上面是具有特殊性的,输出的维度还是会不同,无法并行。(end-start),最终。并行做,难道就不行吗?原创 2023-10-04 11:30:22 · 422 阅读 · 0 评论 -
(Pytorch)判断一个tensor中是否存在nan
有的时候写代码会遇到nan的情况,遇到这种情况你肯定要去解决的,说明你代码有问题。如果你不解决,大概率由于模型前向传播+训练导致模型中所有tensor都是nan,甚至中途就报错了。我们首要是对一些你认为关键的tensor变量进行检测是否为nan,然后停下来,进一步分析nan的原因。nan的原因有非常之多,这里就不介绍了,大家可以看看。原创 2023-09-13 11:49:19 · 1348 阅读 · 0 评论