LongTensor(idx) c = c.index_select(0, idx) vals = torch.cat((c, r[1:])) out_shp = len(c), len(r) n = vals.stride(0) return ...
確定! 回上一頁