摘要: repeat import torch x = torch.tensor([1, 2, 3]) print(x) print(x.repeat(4, 2)) print(x.repeat(4, 2, 1)) permute: 将对应维度的张量置换,有点numpy.transpose()的意思,具体可 阅读全文
posted @ 2020-06-08 11:00 熊猫blue 阅读(168) 评论(0) 推荐(0) 编辑