tensor_nd_products.c 文件参考

N 维张量的轴收缩与模 n 乘积。 更多...

#include "internal/tensor_nd_internal.h"

tensor_nd_products.c 的引用(Include)关系图:

浏览该文件的源代码.

结构体
struct	lmmc_tensor_nd_contraction_t

函数
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)
	模-n 乘积:沿指定模与矩阵收缩.

详细描述

N 维张量的轴收缩与模 n 乘积。

在文件 tensor_nd_products.c 中定义.

---

结构体说明

lmmc_tensor_nd_contraction_t

struct lmmc_tensor_nd_contraction_t

在文件 tensor_nd_products.c 第 7 行定义.

lmmc_tensor_nd_contraction_t 的协作图:

成员变量
size_t	naxes
size_t	contract_size
const size_t *	axes_a
const size_t *	axes_b
size_t	a_free_axes[LMMC_TENSOR_MAX_NDIM]
size_t	b_free_axes[LMMC_TENSOR_MAX_NDIM]
size_t	n_a_free
size_t	n_b_free

函数说明

lmmc_tensor_nd_contract()

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 )

张量收缩:沿匹配轴对求和.

在文件 tensor_nd_products.c 第 196 行定义.

引用了 lmmc_tensor_nd_contraction_t::axes_a, lmmc_tensor_nd_contraction_t::axes_b, LMMC_STATUS_OK, LMMC_TENSOR_MAX_NDIM, lmmc_tensor_nd_contract() , 以及 lmmc_tensor_nd_contraction_t::naxes.

被这些函数引用 lmmc_tensor_nd_contract().

函数调用图:

这是这个函数的调用关系图:

lmmc_tensor_nd_mode_n_product()

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 )

模-n 乘积:沿指定模与矩阵收缩.

在文件 tensor_nd_products.c 第 282 行定义.

引用了 lmmc_tensor_nd_t::dims, LMMC_STATUS_OK, LMMC_TENSOR_MAX_NDIM, lmmc_tensor_nd_mode_n_product(), lmmc_tensor_nd_t::ndim , 以及 lmmc_mat_t::rows.

被这些函数引用 lmmc_tensor_nd_mode_n_product().

函数调用图:

这是这个函数的调用关系图: