pytorch的函数（如squeeze,permute,flatten,stack,numel,matmul）