images = images.expand(-1, 3, -1, -1)
(64, 1, 28, 28)→(64, 3, 28, 28)
将单通道的mnist image 拓展为三通道
猜你喜欢
转载自blog.csdn.net/jining11/article/details/112135231
今日推荐
周排行