PyTorch中tensor.repeat()的使用

學習這個函數的功能的時候,最好仍是要觀察所獲得的結果的維度。很少說,看代碼:python >>> import torch >>> >>> a = torch.randn(33, 55) >>> a.size() torch.Size([33, 55]) >>> >>> a.repeat(1, 1).size() torch.Size([33, 55]) >>> >>> a.repeat(2
相關文章
相關標籤/搜索