matlab单位张量identity tensor创建

identity tensor就是除了所有下标都相等的元素为1以外，其它的元素都为0.

tensor toolbox提供的单位张量并不是该定义下的张量，所以我重新写了个，输入为order阶dimension维度，就获得相对应的单位张量。

my_teneye.m文件

function[I]=my_teneye(order,dimension)
    size1 = repelem(dimension,order);
    I = zeros(size1);
    for i = 1:dimension
        size2 = num2cell(repelem(i,order)); 
        I(size2{:})=1;
    end
    I = tensor(I);
end

命令行测试

my_teneye(4,3)