Pytorch: conv_chain