标签： input precision channels mixed
在有关混合训练精度的nvidia文档中，规定输入通道应为8的倍数以启用张量核心。建议第一层通常具有1（灰度）或3（rgb）的通道，以将通道填充到C = 4且步幅为2。应该如何使用输入张量的格式来进行混合精度训练？使用张量流代码举一个例子