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
相关文章
相关标签/搜索