求多个torch tensor的平均值

  统计/机器学习 Python    浏览次数:5797        分享
0

我有一串torch tensor,shape都是一样的,假设都是5x3的,我怎么对它们求均值呢?就是各个位置的均值,最后得到的还是一个5x3的torch tensor。

假设我有四个tensor,t1, t2, t3, t4。我进行下面的操作

torch.mean([t1, t2, t3, t4], 0)

结果报错

TypeError: mean() received an invalid combination of arguments - got (list, int), but expected one of:
 * (Tensor input)
 * (Tensor input, torch.dtype dtype)
      didn't match because some of the arguments have invalid types: (list, int)
 * (Tensor input, int dim, torch.dtype dtype, Tensor out)
 * (Tensor input, int dim, bool keepdim, torch.dtype dtype, Tensor out)
 * (Tensor input, int dim, bool keepdim, Tensor out)


 

zzzz   2019-05-23 11:45



   1个回答 
3

需要stack之后再mean

torch.mean(torch.stack([t1, t2, t3, t4]), 0)
SofaSofa数据科学社区DS面试题库 DS面经

xiaosu   2019-05-23 14:53

多谢大神 - zzzz   2019-05-24 10:06


  相关主题

怎么把一个大的dataframe按行分割成若干个小的dataframe。   1回答

怎么对有多个空格的字符串进行split?   2回答

怎么check dataframe 中的某个元素是否字符串?   1回答

python __del__ 求解释   3回答

python倒序的range   1回答

python怎么读取txt格式的数据文件?   1回答

怎么把python的dict存为json格式文件   2回答

python里的continue是什么意思   2回答

print里的"%.2f"是什么意思?   1回答

有一个1000万行的dataframe   3回答

from __future__ import division是什么意思   3回答

python怎么去除字符串中的连字符?   1回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!