技术栈
拆分组合
qq_41920323
7 小时前
python
·
torch
·
tensor
·
拆分组合
tensor向量按任意维度进行切片、拆分、组合
torch.index_select(input_tensor, 切片维度, 切片索引)注意：切完之后，转onnx时会生成Gather节点；