RuntimeError：未针对类型 torch.ByteTensor 实现均值

【问题标题】：RuntimeError: mean is not implemented for type torch.ByteTensorRuntimeError：未针对类型 torch.ByteTensor 实现均值
【发布时间】：2021-06-09 15:48:47
【问题描述】：

运行代码后出现此错误： “RuntimeError：没有为类型torch.ByteTensor实现平均值”？有谁知道我在这里做错了什么？

accuracy = torch.mean(output)

【问题讨论】：

标签： neural-network pytorch

【解决方案1】：

知道了，基本上torch.mean() 没有在torch.ByteTensor 上实现，所以我们可以将它转换为torch.mean() 支持的FloatTensor。所以代码会变成：

accuracy = torch.mean(output.type(torch.FloatTensor))

【讨论】：