使用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刪除。