2022年4月18日
【pytorch】对numpy对象进行增加维度
摘要： 使用unsqueeze()来增加维度：x = torch.from_numpy(img).float().unsqueeze(0)，其中的参数0是指“在第0个维度增加一维” 如果想再增加一个维度，则只需要再添加一行unsqueeze(0)即可 阅读全文
posted @ 2022-04-18 17:24 ShengOasis 阅读(1027) 评论(0) 推荐(0)
【jupyter】使用plt画图时，jupyter内核崩溃
摘要： 报错情况： 解决： 在pytorch中导入以下模块： import os os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE" 重新运行即可 阅读全文
posted @ 2022-04-18 10:43 ShengOasis 阅读(982) 评论(0) 推荐(0)