Method gather

src/python/lfs/py_tensor.cpp:1257–1259

    }

    PyTensor PyTensor::gather(int dim, const PyTensor& indices) const {
        return PyTensor(tensor_.gather(dim, indices.tensor_));
    }

    PyTensor PyTensor::masked_select(const PyTensor& mask) const {
        return PyTensor(tensor_.masked_select(mask.tensor_));