2019年11月25日
PyTorch笔记之 squeeze() 和 unsqueeze()
摘要： squeeze() 函数 squeeze() 用来去掉向量的一个维度，只有维度为 1 的那一维才能去掉 example： 初始化1个向量shape为（1,2,3）的向量 import torch a = torch.rand((1,2,3)) 去掉第 0 维，第 0 维的大小是1，所以可以去掉第0维 阅读全文
posted @ 2019-11-25 21:08 那少年和狗 阅读(3723) 评论(0) 推荐(1)