摘要：chatgpt回答： 要得到一个形状为 [7, 6, 49, 3] 的 NumPy 矩阵，你需要组合这两个原始的矩阵（假设为矩阵 A 和矩阵 B）。不过，由于这两个矩阵的形状分别是 [7, 49, 2] 和 [6]，它们无法直接通过常规的广播规则来获得一个 [7, 6, 49, 3] 形状的矩阵。但 阅读全文

摘要：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 阅读全文

摘要：转载于：https://www.zhihu.com/question/562282138/answer/2947708508?utm_id=0 官方文档链接： https://pytorch.org/docs/stable/generated/torch.gather.html#torch.gath 阅读全文