RuntimeError: _th_histc not supported on CPUType for Long

count = torch.histc(target, bins=10, min=0, max=9)

代码报错是因为 target 的数据类型为整型,改为浮点型就可以了。

改正:

count = torch.histc(target.float(), bins=10, min=0)

你可能感兴趣的:(三两闲事,python)