PyTorch学习 | squeeze()和unsqueeze()函数

squeeze 减少维度

unsqueeze增加维度

expand 扩展维度

https://blog.csdn.net/flysky_jay/article/details/81607289

posted @ 2021-11-14 10:33  !harper  阅读(312)  评论(0)  编辑  收藏  举报