Get values along an axis given an indices vector (using index_select) t = torch. Tensor([[1,2], [3,4], [5,6]]) index = torch. Long Tensor([0,0]) print ...
確定! 回上一頁