pytorch中的repeat()

这两天写程序中须要,查到了一个很是好用的函数,烂笔头比好记性好用,记录下:python import torch x = torch.tensor([1, 2, 3]) print(x.repeat(4)) print("###################################") print(x.repeat(4, 1)) 运行结果以下: web import torch x =
相关文章
相关标签/搜索