- 博客(3)
- 收藏
- 关注
原创 python assert
python assert assert 作用为:判断一个表达式是否正确,若返回值为false,触发异常,并输出自己设置的提示信息。 style_img = image_loader("data/starry_night.jpg") content_img = image_loader("data/shanghai_buildings.bmp") print("style size:", style_img.size()) # style size: torch.Size([1, 3, 12
2021-03-02 10:38:42
183
原创 Python squeeze
Python Squeeze squeeze 移除数组中维度为1的维度 def forward(self, x, labels): x = x.view(x.size(0), 784) # (100,784) c = self.label_emb(labels) x = torch.cat([x, c], 1) # (100, 794) out = self.model(x) # (100, 1)
2021-03-01 14:59:28
456
原创 Transforms ToTensor与Normalize 2021-03-01
Transformstransforms提供了对PIL Image对象和Tensor对象的常用操作ToTensorNormalize实例 transforms提供了对PIL Image对象和Tensor对象的常用操作 ToTensor 将一个取值范围在[0,255]的PIL.Image转换为Tensor。形状为(H,W,C)的Numpy.ndarry转换为形状[C,H,W],取值范围是[0,1.0]的torch.FloatTensor。 Normalize 即进行标准化操作。 image = (image
2021-03-01 10:23:04
340
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅