理解：2个样本数据，每个数据3行(T_x = 3)，4列(n_value = 4)
indices = np.argmax(pred, 2)
例如，对第一个样本数据的，第一行中的，所有数据取最大值的索引下标
3维numpy数组理解和np.argmax(a, axix=2)
猜你喜欢
转载自www.cnblogs.com/douzujun/p/13199880.html
今日推荐
周排行