Pytorch中torch.mean()函数
Pytorch中torch.mean()函数TORCH.MEAN参考了官方文档的详细介绍。基本语法:torch.mean(Tensor) 函数返回了一个输入张量Tensor中所有元素的平均值,返回值同样是tensor类型。具体的使用中,该函数的形参有两种:1、可以是一个tensor;2、也有另一种重载形式。对于第一种情况,官方文档给出的示例很好理解,形参列表输入tensor类型的张量,输出均值te
·
Pytorch中torch.mean()函数
TORCH.MEAN
参考了官方文档的详细介绍。
基本语法:
torch.mean(Tensor) 函数返回了一个输入张量Tensor中所有元素的平均值,返回值同样是tensor类型。
具体的使用中,该函数的形参有两种:
1、可以是一个tensor;
2、也有另一种重载形式。
对于第一种情况,官方文档给出的示例很好理解,形参列表输入tensor类型的张量,输出均值tensor。
例如:
>>> a = torch.randn(1, 3)
>>> a
tensor([[ 0.2294, -0.5481, 1.3288]])
>>> torch.mean(a)
tensor(0.3367)
对于第二种情况,形参列表有所改变。
基本语法:
返回了输入input(输入是tensor类型)的特定维度dim上的均值,如果dim是一个维度的列表,参考如下博客里的详细讲解
注意:
对tensor使用mean操作时,需要转换数据类型,如果使用int型会报错。RuntimeError: Can only calculate the mean of floating types. Got Long instead
本文介绍的内容,针对tensor.sum()等函数同样适用。
参考链接:https://pytorch.org/docs/stable/generated/torch.mean.html#torch.mean
更多推荐
已为社区贡献1条内容
所有评论(0)