2025年3月8日
张量的索引
摘要： 整数列表选择 x = torch.tensor([[[10, 20, 30], [40, 50, 60]]]) x1 = x[[0, 0]] x2 = x[[0, 0], [0, 1]] x3 = x[0, [0, 1], [0, 1]] print(x1) print(x2) print(x3) 阅读全文
posted @ 2025-03-08 19:23 xiezhengcai 阅读(19) 评论(0) 推荐(0)