摘要: M1,M2,M3是形状相同(任意维度和形状都可以,记为shape)的index matrix。 M1的元素是0,1;M2取值于[0:8],M3取值于[0:768] 在data的形状为torch.Size([2, 8, 768])的数据中,取出data[M1,M2,M3]。其形状就是shape,其实就 阅读全文
posted @ 2021-11-16 12:49 zae 阅读(310) 评论(0) 推荐(0) 编辑
摘要: torch.arange((10))[None, :],torch.arange((10))[:,None] #在外面增加了一个维度 output: (tensor([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]]), tensor([[0], [1], [2], [3], [4], 阅读全文
posted @ 2021-11-16 11:40 zae 阅读(80) 评论(0) 推荐(0) 编辑