uc-modern-cpp-student/cc/tensor/tensor.cc

#include "tensor.h"

namespace tensor {

std::shared_ptr<Tensor> Tensor::transpose() {
    // 放心，下面的代码暂时不会被触发，我们假定所有的tensor都是2维的
    // if (shape.size() != 2) {
    //     throw std::runtime_error("Transpose is only supported for 2D tensors.");
    // }

    // 这里能够获得矩阵的行数和列数，但是我们是使用一个一维的vector来存储数据的。该如何实现“转置”呢？
    std::size_t rows = shape[0];
    std::size_t cols = shape[1];
    std::vector<size_t> new_shape = {cols, rows};
    // 你知道这里的size变量在哪里定义的吗？在VSCode里面安装C/C++ Extension Pack后，按下ctrl键并单击变量size，VSCode就会把你导向定义这个变量的地方！
    std::vector<float> transposed_data(size);

    // 请在这里写转置的代码

    // 请阅读关于Tensor的定义，在这里创建一个新的Tensor
    // 注意，要使用shared_ptr哦！
    
    return std::make_shared<Tensor>(new_shape);
}


std::shared_ptr<Tensor> pyarray_to_tensor(py::array_t<float> array) {
    py::buffer_info info = array.request();
    float* dataPtr = static_cast<float*>(info.ptr);
    std::vector<std::size_t> shape = {};
    for (auto &it: info.shape) {
        shape.push_back(it);
    }
    auto tensor = std::make_shared<Tensor>(shape);
    std::vector<float> result(dataPtr, dataPtr + info.size);
    tensor->data = result;
    return tensor;
}

std::shared_ptr<Tensor> argmax(const std::shared_ptr<Tensor>& tensor, int axis) {
    // you only need to handle the two dimensional tensor, and the axis can be either 0 or 1
    // the tensor's shape is (batch_size, features)
    // if the axis is 0, it outputs a tensor (1, features)
    // if the axis is 1, it outputs a tensor (batch_size, 1)

    // compute the output's shape
    std::vector<std::size_t> output_shape = tensor->shape;
    output_shape.erase(output_shape.begin() + axis);

    auto result = std::make_shared<Tensor>(output_shape);
    // 这个问题似乎有点难，所以我们决定给你送点分。一个简单的办法是分axis为0还是为1来进行讨论，反正我们已经把问题简化为了，在一个二维的tensor里面，找到每一行或者每一列的最大值，并输出一个一维的tensor。
    // 补全这里的代码。
    return result;
}

std::shared_ptr<Tensor> mean(const std::shared_ptr<Tensor>& tensor) {
    std::vector<std::size_t> shape = {1};
    auto result = std::make_shared<Tensor>(shape);
    auto sum = 0.0f;
    for (auto &it: tensor->data) {
        sum += it;
    }
    sum /= tensor->size;
    result->data[0] = sum;
    return result;
}

std::shared_ptr<Tensor> exp(const std::shared_ptr<Tensor>& tensor) {
    auto result = std::make_shared<Tensor>(tensor->shape);
    for (auto i = 0; i < tensor->size; i++) {
        result->data[i] = expf(tensor->data[i]);
    }
    return result;
}

}