tensorflow和pytorch二维卷积的数据格式