使用unsqueeze()来增加维度：x = torch.from_numpy(img).float().unsqueeze(0)，其中的参数0是指“在第0个维度增加一维”
如果想再增加一个维度，则只需要再添加一行
unsqueeze(0)即可
【pytorch】对numpy对象进行增加维度
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。
使用unsqueeze()来增加维度：x = torch.from_numpy(img).float().unsqueeze(0)，其中的参数0是指“在第0个维度增加一维”
如果想再增加一个维度，则只需要再添加一行
unsqueeze(0)即可
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。