logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

Sizes of tensors must match except in dimension 1. Expected size 44 but got size 45 for tensor numbe

项目场景:跑U-net网络的时候,有一步是torch.cat()操作,出现下面是代码class UNet(nn.Module):def __init__(self, in_channels, out_channels):super(UNet, self).__init__()self.conv1 = DoubleConv(in_channels, 32)self.pool1 = nn.MaxPoo

#深度学习#pytorch#python
暂无文章信息