Method stack

src/python/lfs/py_tensor.cpp:1575–1582

    }

    PyTensor PyTensor::stack(const std::vector<PyTensor>& tensors, int dim) {
        std::vector<Tensor> core_tensors;
        core_tensors.reserve(tensors.size());
        for (const auto& t : tensors) {
            core_tensors.push_back(t.tensor_);
        }
        return PyTensor(Tensor::stack(core_tensors, dim));
    }

    PyTensor PyTensor::where(const PyTensor& condition, const PyTensor& x, const PyTensor& y) {
        return PyTensor(Tensor::where(condition.tensor_, x.tensor_, y.tensor_));