判断输入是否含nan

2020-02-17  本文已影响0人  发现一个喜悦的地方

if np.any(np.isnan(input.cpu().numpy())):
print('Input data has NaN!')

上一篇下一篇

猜你喜欢

热点阅读