3.Reduction operations
1.我们经常在神经网络输出预测张量上使用Argmax函数,这让我们可以确定哪个类别的预测值最高。
这是因为输出张量的每个指标都对应于一个特定的预测类别。
1.看看这些操作,当我们称它们为3x3张量的均值时,减少的输出是一个标量张量。
2.如果我们想要把这个值作为一个数字,我们可以使用item(),它只适用于标量张量。
1.我们经常在神经网络输出预测张量上使用Argmax函数,这让我们可以确定哪个类别的预测值最高。
这是因为输出张量的每个指标都对应于一个特定的预测类别。
1.看看这些操作,当我们称它们为3x3张量的均值时,减少的输出是一个标量张量。
2.如果我们想要把这个值作为一个数字,我们可以使用item(),它只适用于标量张量。