torch.mean是对torch.Tensor取均值。具体怎么使用,请看下面的例子. 不指定维度求均值. 如果不指定维度,是对Tensor里所有元素求均值,最后计算出来的结果是一个标量。
確定! 回上一頁