pytorch中dataSet里的__getitem__()可以返回numpy类型,

pytorch中dataSet里的__getitem__()可以返回numpy类型,自动转为torch.tensor

发布了51 篇原创文章 · 获赞 29 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qwer7512090/article/details/103452174