Method tolist

src/python/lfs/py_tensor.cpp:468–502

    }

    nb::object PyTensor::tolist() const {
        validate();
        Tensor cpu_tensor = tensor_.device() == Device::CUDA ? tensor_.cpu() : tensor_;
        if (!cpu_tensor.is_contiguous()) {
            cpu_tensor = cpu_tensor.contiguous();
        }

        const auto& dims = cpu_tensor.shape().dims();
        size_t offset = 0;

        switch (cpu_tensor.dtype()) {
        case DataType::Float32: {
            const auto values = cpu_tensor.to_vector();
            return build_nested_list(dims, 0, offset, [&](size_t index) { return nb::cast(values[index]); });
        }
        case DataType::Int32: {
            const auto values = cpu_tensor.to_vector_int();
            return build_nested_list(dims, 0, offset, [&](size_t index) { return nb::cast(values[index]); });
        }
        case DataType::Int64: {
            const auto values = cpu_tensor.to_vector_int64();
            return build_nested_list(dims, 0, offset, [&](size_t index) { return nb::cast(values[index]); });
        }
        case DataType::UInt8: {
            const auto values = cpu_tensor.to_vector_uint8();
            return build_nested_list(dims, 0, offset, [&](size_t index) { return nb::cast(values[index]); });
        }
        case DataType::Bool: {
            const auto values = cpu_tensor.to_vector_bool();
            return build_nested_list(dims, 0, offset, [&](size_t index) { return nb::cast(static_cast<bool>(values[index])); });
        }
        default:
            throw std::runtime_error("Unsupported dtype for tolist conversion");
        }
    }

    size_t PyTensor::count_nonzero() const {
        validate();