在帶有 PyTorch 的 Python 中,如果你有一個數組:torch.linspace(0, 10, 10)你可以只使用前三個元素,例如reduced_tensor = torch.linspace(0, 10, 10)[:4].C++/libtorch 中是否有類似于數組切片的模擬[:]?如果沒有,我怎樣才能輕松實現這一目標?
添加回答
舉報
0/150
提交
取消