第七色在线视频,2021少妇久久久久久久久久,亚洲欧洲精品成人久久av18,亚洲国产精品特色大片观看完整版,孙宇晨将参加特朗普的晚宴

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

如何對火炬張量應用變換

如何對火炬張量應用變換

郎朗坤 2023-07-18 10:20:05
我有一個火炬張量z,我想應用一個變換矩陣,mat并使z輸出的大小與 完全相同z。這是我正在運行的代碼:def trans(z):    print(z)    mat = transforms.Compose([transforms.ToPILImage(),transforms.RandomRotation(90),transforms.ToTensor()])    z = Variable(mat(z.cpu()).cuda())    z = nnf.interpolate(z, size=(28, 28), mode='linear', align_corners=False)    return zz = trans(z)但是,我收到此錯誤:RuntimeError                              Traceback (most recent call last)<ipython-input-12-e2fc36889ba5> in <module>()      3 inputs,targs=next(iter(tst_loader))      4 recon, mean, var = vae.predict(model, inputs[img_idx])----> 5 out = vae.generate(model, mean, var)4 frames/content/vae.py in generate(model, mean, var)     90     z = trans(z)     91     z = Variable(z.cpu().cuda())---> 92     out = model.decode(z)     93     return out.data.cpu()     94 /content/vae.py in decode(self, z)     56      57     def decode(self, z):---> 58         out = self.z_develop(z)     59         out = out.view(z.size(0), 64, self.z_dim, self.z_dim)     60         out = self.decoder(out)/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py in _call_impl(self, *input, **kwargs)    720             result = self._slow_forward(*input, **kwargs)    721         else:--> 722             result = self.forward(*input, **kwargs)    723         for hook in itertools.chain(    724                 _global_forward_hooks.values(),/usr/local/lib/python3.6/dist-packages/torch/nn/modules/linear.py in forward(self, input)     89      90     def forward(self, input: Tensor) -> Tensor:---> 91         return F.linear(input, self.weight, self.bias)     92      93     def extra_repr(self) -> str:/usr/local/lib/python3.6/dist-packages/torch/nn/functional.py in linear(input, weight, bias)   1674         ret = torch.addmm(bias, input, weight.t())   1675     else:-> 1676         output = input.matmul(weight.t())   1677         if bias is not None:   1678             output += biasRuntimeError: mat1 dim 1 must match mat2 dim 0如何成功應用此旋轉(zhuǎn)變換mat并且不會出現(xiàn)任何錯誤?
查看完整描述

1 回答

?
蝴蝶刀刀

TA貢獻1801條經(jīng)驗 獲得超8個贊

問題是interpolate需要一個批次維度,但根據(jù)錯誤消息和transforms.?由于您的輸入是空間的(基于size=(28, 28)),您可以通過添加批量維度并更改 來解決該問題mode,因為linear沒有針對空間輸入實現(xiàn):

z?=?nnf.interpolate(z.unsqueeze(0),?size=(28,?28),?mode='bilinear',?align_corners=False)

如果你z仍然想擁有像 (C, H, W) 這樣的形狀,那么:

z?=?nnf.interpolate(z.unsqueeze(0),?size=(28,?28),?mode='bilinear',?align_corners=False).squeeze(0)


查看完整回答
反對 回復 2023-07-18
  • 1 回答
  • 0 關注
  • 115 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯(lián)系客服咨詢優(yōu)惠詳情

幫助反饋 APP下載

慕課網(wǎng)APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網(wǎng)微信公眾號