C++泛型 重载 <<

#ifndef VECTOR_OPS_FOR_NN
#define VECTOR_OPS_FOR_NN

#include <algorithm>
#include <chrono>
#include <iostream>
#include <random>
#include <valarray>
#include <vector>

namespace machine_learning{
    // 重载<< 打印二维数组
    template <typename T>
    std::ostream &operator<<(std::ostream &out, std::vector<std::valarray<T>> const &A){
        out.precision(4);   // 取数据的前四位有效数字
        for(const auto &a : A){
            for(const auto &x : a){
                std::cout << x << ' ';
            }
            std::cout << std::endl;
        }
        return out;
    }

    // 重载 << 打印pair
    template <typename T>
    std::ostream &operator <<(std::ostream &out, const std::pair<T,T> &A){
        out.precision(4);
        // printing pair in the form (p, q)
        std::cout << "(" << A.first << ", " << A.second << ")";
        return out;
    }

    // 重载<< 打印一维数组
    template <typename T>
    std::ostream &operator<<(std::ostream &out, std::vector<T> const &A){
        out.precision(4);
        for(const auto &a : A){
            std::cout << a << ' ';
        }
        std::cout << std::endl;
        return out;
    }

    template <typename T>
    std::ostream &operator<<(std::ostream &out, std::valarray<T> const &A){
        out.precision(4);
        for(const auto &a : A){
            std::cout << a << ' ';
        }
        std::cout << std::endl;
        return out;
    }

    // 将element插入1D vector
    template <typename T>
    std::valarray<T> insert_element(const std::valarray<T> &A, const T &ele){
        std::valarray<T> B;     // 存储新的1D
        B.resize(A.size() + 1); // 调整大小
        for (size_t i = 0; i < A.size(); i++){
            B[i] = A[i];
        }
        B[B.size() - 1] = ele;  // 最后位置插入新元素
        return B;
    }


    void test01(){


        // print 1D vector
        std::vector<int> weekdays;
        weekdays.push_back(2);
        weekdays.push_back(3);
        weekdays.push_back(4);
        weekdays.push_back(5);
        std::cout << "print 1D vector: " << weekdays;

        // insert element
        // 创建一个大小为4、值全为3的valarray
        std::valarray<int> valarray_A(3, 4);
        int a = 5;
        std::valarray<int> B = insert_element(valarray_A, a);
        std::cout << "insert_element: " << B << std::endl;

        // print vector<std::valarray<int>>
        std::vector<std::valarray<int>> vec_va;
        vec_va.push_back(B);
        vec_va.push_back(B);
        std::cout << "print vector<std::valarray<int>>: " << vec_va << std::endl;

        // print pair
        std::pair<int, int> pair_A = {1, 2};
        std::cout << "print pair = "<< pair_A << std::endl;
    }
}
#endif

int main(){
    machine_learning::test01();
}

my github

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

DYF-AI

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值