
【PyTorch学习记录】
记录Pytorch使用过程中的基础知识
下雨天不下雨
深耕于计算机视觉、目标检测、语义分割、遥感图像处理、医学图像处理等领域的菜鸟一枚,请多多指教。
展开
-
【PyTorch基础】——expand()和expand_as()
返回输入张量在某维扩展为更大尺寸后的张量,且原始tensor和扩展后tensor不共享内存。括号中输入参数为指定经过维度尺寸扩展后的张量的size。expand()函数只能将size=1的维度扩展到更大的尺寸,如果扩展其他size()的维度会报错。(如下:示例三)原创 2022-12-08 21:57:58 · 3192 阅读 · 0 评论 -
【PyTorch基础】——nn.Conv2d()参数设置
nn.Conv2d()参数设置bias什么时候设置为True?什么时候设置为False?原创 2022-12-03 14:34:14 · 3706 阅读 · 0 评论