2022年9月5日
torch.repeat()
摘要： PyTorch中的repeat()函数可以对张量进行重复扩充。 >>> a= torch.arange(24).reshape(1,2,3,4) >>> print(a.size()) torch.Size([1, 2, 3, 4]) >>> print("b.size",a.repeat(2,2, 阅读全文
posted @ 2022-09-05 17:19 魏晋南北朝 阅读(432) 评论(0) 推荐(0) 编辑