[Pytorch]何为tensor(二)

3.Reduction operations

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

4.Access operations(略)

猜你喜欢

转载自blog.csdn.net/weixin_66896881/article/details/128620630