Method num_elements

src/framework/core/module.cpp:76–82

}

int64_t TensorShape::num_elements() const {
    int64_t total = 1;
    for (size_t i = 0; i < rank; ++i) {
        total *= dims[i];
    }
    return total;
}

TensorShape TensorShape::with_last_dim(int64_t value) const {
    TensorShape result = *this;