1、加法运算A=torch.arange(20,dtype=torch.float32).reshape(5,4)B=A.clone()#通过分配新内存,将A的一个副本分配给BA,A+B#tensor([[0.,1.,2.,3.],#[4.,5.,6.,7.],#[8.,9.,10.,11.],#[12.,13.,14.,15.],#[16.,17.,18.,19.]]),#tensor([[0.,2.,4.,6.],#[8.,10.,12.,14.],#[16.,18.,20.,22.],#[24.,26.,28.,30.],#[32.,34.,36.,38.]])2、乘法运算A*B#ten
🍨本文为🔗365天深度学习训练营 中的学习记录博客🍦参考文章:Pytorch实战|第P8天:YOLOv5-C3模块实现(训练营内部成员可读)🍖原作者:K同学啊|接辅导、项目定制了解C3的结构,方便后续YOLOv5算法的学习。采用的数据集是天气识别的数据集。 一、前期准备1.设置GPUimporttorchimporttorch.nnasnnimporttorchvision.transformsastransformsimporttorchvisionfromtorchvisionimporttransforms,datasetsimportos,PIL,pathlib,warningswar