2024年2月19日
repeat得到的是[b0 b1 b0 b1]现在需要[b0 b0 b1 b1]
摘要： pytorch 一个tensor 比如是 [b0 b1 ] 用tensor.repeat(2)函数可以得到 [b0 b1 b0 b1 ] 我现在想得到 [b0 b0 b1 b1 ] 如何优雅的得到？ import torch c = torch.randint(0, 9, (2, 3)) d = c 阅读全文
posted @ 2024-02-19 16:48 无左无右 阅读(4) 评论(0) 推荐(0) 编辑