meter.py 文件源码

def to_numpy(x):
    if isinstance(x, Variable):
        return x.data.cpu().squeeze().numpy()
    elif torch.is_tensor(x):
        return x.cpu().squeeze().numpy()
    else:
        return x


# reference: https://github.com/pytorch/tnt/blob/master/torchnet/meter/msemeter.py