摘要: 主要用于记录一些比较常用的函数,便于查找。未记录的可查看PYTORCH TUTORIALS,PyTorch中文文档 torch.unsqueeze 主要用于在某一位置添加一个维度 x = torch.tensor([1, 2, 3, 4]) torch.unsqueeze(x, 0) torch.u 阅读全文
posted @ 2022-01-30 17:52 xingye_z 阅读(385) 评论(0) 推荐(0) 编辑