tensorflow和PyTorch在卷积层中输入shape的区别

tensorflow输入shape为(batch_size, in_channels, height, weight)

pytorch的输入shape为(batch_size, height, weight, in_channels)

这在两种框架之间进行模型转换时将会很有用

你可能感兴趣的:(python,python)