Can not squeeze dim[1], expected a dimension of 1, got 21

2021-09-28 159 阅读1分钟

这是因为tf的文档要求label必须是一个单独的值，而不是一个one-hot编码后像[1,0,0], [0,1,0], [0,0,1].的值。