2020年11月3日
torch.squeeze函数解释，torch.FloatTensor()函数作用解释
摘要： 1. torch.squeeze(x,N) 主要对数据维度进行压缩 torch.squeeze(x,N) #也可以写为格式 x.squeeze(dim=N) 含义：当N未给定时，去掉x中所有维度为1的维度，当N给定为某一数值时，去掉在这一数值指定位置的维度为1的维度。 代码示例： 2.torch.u 阅读全文
posted @ 2020-11-03 20:00 阳光中的影子 阅读(8768) 评论(0) 推荐(1) 编辑