问题描述
有一个list里面包含三个tensor,想计算它的均值。
解决办法
代码如下:
pre_list = [tensor1,tensor1,tensor3] #数值如上图
pres = torch.stack(pre_list, dim=0).mean(dim=0)
运行结果如下:
成功!
参考链接:https://blog.csdn.net/qq_41670507/article/details/128054358
有一个list里面包含三个tensor,想计算它的均值。
代码如下:
pre_list = [tensor1,tensor1,tensor3] #数值如上图
pres = torch.stack(pre_list, dim=0).mean(dim=0)
运行结果如下:
成功!
参考链接:https://blog.csdn.net/qq_41670507/article/details/128054358