PyTorch CNN linear layer shape after conv2d

后端 未结 0 879
余生分开走
余生分开走 2021-02-01 01:05

I was trying to learn PyTorch and came across a tutorial where a CNN is defined like below,

class Net(Module):   
    def __init__(self):
        super(Net, self)         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题