tensor_nd.h

浏览该文件的文档.

#ifndef LMMC_TENSOR_ND_H
#define LMMC_TENSOR_ND_H
 
#include <stddef.h>
#include "lmmc/dense.h"
#include "lmmc/status.h"
 
#ifdef __cplusplus
extern "C" {
#endif

#define LMMC_TENSOR_MAX_NDIM 8

typedef struct {
    size_t ndim;                            
    size_t dims[LMMC_TENSOR_MAX_NDIM];      
    size_t strides[LMMC_TENSOR_MAX_NDIM];   
    lmmc_real_t* data;                      
    int owns_data;                          
} lmmc_tensor_nd_t;

lmmc_status_t lmmc_tensor_nd_create(size_t ndim, const size_t* dims, lmmc_tensor_nd_t* out);

lmmc_status_t lmmc_tensor_nd_get(const lmmc_tensor_nd_t* t, const size_t* idx, lmmc_real_t* out);

lmmc_status_t lmmc_tensor_nd_set(lmmc_tensor_nd_t* t, const size_t* idx, lmmc_real_t value);

lmmc_status_t lmmc_tensor_nd_permute(const lmmc_tensor_nd_t* in, const size_t* perm, lmmc_tensor_nd_t* out);

lmmc_status_t lmmc_tensor_nd_contract(const lmmc_tensor_nd_t* a, const lmmc_tensor_nd_t* b,
    const size_t* axes_a, const size_t* axes_b, size_t naxes, lmmc_tensor_nd_t* out);

lmmc_status_t lmmc_tensor_nd_mode_n_product(const lmmc_tensor_nd_t* t, const lmmc_mat_t* mat,
    size_t mode, lmmc_tensor_nd_t* out);

lmmc_status_t lmmc_tensor_nd_reshape_view(const lmmc_tensor_nd_t* src,
    size_t new_ndim, const size_t* new_dims, lmmc_tensor_nd_t* out_view);

void lmmc_tensor_nd_destroy(lmmc_tensor_nd_t* t);
 
#ifdef __cplusplus
}
#endif
 
#endif