摘要: squeeze(): squeeze(arg)表示第arg维的维度值为1,则去掉该维度。否则tensor不变。(即若tensor.shape()[arg] = 1,则去掉该维度) unsqueeze(): unsqueeze(arg)与squeeze(arg)作用相反,表示在第arg维增加一个维度值 阅读全文
posted @ 2018-06-26 22:33 sbj123456789 阅读(24393) 评论(0) 推荐(1) 编辑
摘要: torchvision.transforms 先看看一个transform是干啥的: transform通常用于处理图像,它将图像进行一定的变换(transform),具体来说有: class torchvision.transforms.ToTensor 把一个取值范围是[0,255]的PIL.I 阅读全文
posted @ 2018-06-26 20:38 sbj123456789 阅读(537) 评论(0) 推荐(0) 编辑