tensor3.h

浏览该文件的文档.

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

typedef struct {
    size_t dim0;          
    size_t dim1;          
    size_t dim2;          
    size_t stride0;       
    size_t stride1;       
    size_t stride2;       
    lmmc_real_t* data;    
    int owns_data;        
} lmmc_tensor3_t;

lmmc_status_t lmmc_tensor3_create(size_t dim0, size_t dim1, size_t dim2, lmmc_tensor3_t* out_tensor);

lmmc_status_t lmmc_tensor3_wrap(
    size_t dim0,
    size_t dim1,
    size_t dim2,
    size_t stride0,
    size_t stride1,
    size_t stride2,
    lmmc_real_t* data,
    lmmc_tensor3_t* out_tensor
);

void lmmc_tensor3_destroy(lmmc_tensor3_t* tensor);

lmmc_status_t lmmc_tensor3_fill(lmmc_tensor3_t* tensor, lmmc_real_t value);

lmmc_status_t lmmc_tensor3_set(lmmc_tensor3_t* tensor, size_t i, size_t j, size_t k, lmmc_real_t value);

lmmc_status_t lmmc_tensor3_get(const lmmc_tensor3_t* tensor, size_t i, size_t j, size_t k, lmmc_real_t* out_value);

lmmc_status_t lmmc_tensor3_norm_fro(const lmmc_tensor3_t* tensor, lmmc_real_t* out_norm);

lmmc_status_t lmmc_tensor3_add(const lmmc_tensor3_t* a, const lmmc_tensor3_t* b, lmmc_tensor3_t* out_tensor);
lmmc_status_t lmmc_tensor3_sub(const lmmc_tensor3_t* a, const lmmc_tensor3_t* b, lmmc_tensor3_t* out_tensor);
lmmc_status_t lmmc_tensor3_mul(const lmmc_tensor3_t* a, const lmmc_tensor3_t* b, lmmc_tensor3_t* out_tensor);
lmmc_status_t lmmc_tensor3_div(const lmmc_tensor3_t* a, const lmmc_tensor3_t* b, lmmc_tensor3_t* out_tensor);
lmmc_status_t lmmc_tensor3_scale(const lmmc_tensor3_t* tensor, lmmc_real_t alpha, lmmc_tensor3_t* out_tensor);

lmmc_status_t lmmc_tensor3_sum(const lmmc_tensor3_t* tensor, lmmc_real_t* out_sum);
lmmc_status_t lmmc_tensor3_max(const lmmc_tensor3_t* tensor, lmmc_real_t* out_max);
lmmc_status_t lmmc_tensor3_min(const lmmc_tensor3_t* tensor, lmmc_real_t* out_min);

lmmc_status_t lmmc_tensor3_sum_axis(const lmmc_tensor3_t* tensor, size_t axis, lmmc_mat_t* out_matrix);

lmmc_status_t lmmc_tensor3_reshape_view(
    const lmmc_tensor3_t* tensor,
    size_t new_dim0,
    size_t new_dim1,
    size_t new_dim2,
    lmmc_tensor3_t* out_view
);

lmmc_status_t lmmc_tensor3_slice_view(
    const lmmc_tensor3_t* tensor,
    size_t begin0,
    size_t end0,
    size_t begin1,
    size_t end1,
    size_t begin2,
    size_t end2,
    lmmc_tensor3_t* out_view
);
 
#ifdef __cplusplus
}
#endif
 
#endif