RuntimeError: Can only calculate the mean of floating types. Got Long instead.

最新推荐文章于 2023-07-22 07:15:00 发布

Encounter84

最新推荐文章于 2023-07-22 07:15:00 发布

阅读量511

点赞数 1

文章标签： python 开发语言

本文链接：https://blog.csdn.net/zhuzyibooooo/article/details/131037781

版权

遇到这种问题时，我是使用了torch.mean函数。如果不添加a = a.float()，会报这个错。原因是因为mean函数只能对float类型数据计算平均值，而a是int64类型。因此将其转换为float类型即可。

import torch

a = torch.randint(1,9,[3,4])
a = a.float()
print(a.shape)
print(a)

zeros = torch.zeros_like(a[0,:])
ones = torch.ones_like(a[0,:])
output = a.clone()

for n in range(a.shape[0]):
    output[n, :] = torch.where(a[n, :] >= torch.mean(a[n, :]), ones, zeros)
print(output.shape)
print(output)

顺便介绍一下torch.mean()函数，mean()函数的参数：dim=0,按列求平均值，返回的形状是（1，列数）；dim=1,按行求平均值，返回的形状是（行数，1）,默认不设置dim的时候，返回的是所有元素的平均值。