摘要:
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,其实就 阅读全文
摘要:
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], 阅读全文