Tensor-Matrix互相转换 Python版

最新推荐文章于 2024-07-05 15:18:23 发布

Cute_Abacus

最新推荐文章于 2024-07-05 15:18:23 发布

阅读量1.5k

点赞数

分类专栏：基本知识文章标签： python 矩阵

本文链接：https://blog.csdn.net/LUCKY_SANTA/article/details/125105128

版权

本文介绍如何在Python中将三阶张量转换为矩阵，内容包括张量展开的概念，并分享了从Matlab代码转化而来的Python实现。

摘要由CSDN通过智能技术生成

先验知识：张量可以按照不同阶展开成不同的矩阵
具体可以参考：如何加单理解和实现张量展开

但是查到的资料都不是python的，我根据matlab版本的代码做了一些转换
在这里共享给大家（目前应该只适用三阶张量）

# Author--Zoe
# 2022/06/02

'''
tensor 张量
mat	矩阵
mode 分解的阶
tensor_size 张量每维的大小构成的数组
'''
# 张量转矩阵
def ten2mat(tensor, mode):
    return np.reshape(np.moveaxis(tensor, mode, 0), (tensor.shape[mode], -1), order = 'F')

# 矩阵转张量
def mat2ten(mat, tensor_size, mode):
    index = list()
    index