gpt4 book ai didi

julia - 如果我不知道总维数,如何仅按一维索引数组?

转载 作者:行者123 更新时间:2023-12-02 09:07:56 24 4
gpt4 key购买 nike

我有一个(可能是多维的)数组X。它至少有 k 维度。我不知道它有多少维度。我如何仅按 k 维度对其进行索引?

例如如果X是4维的,k是3,想要的索引是7,我想得到X[:, :, 7, :].

最佳答案

使用selectdim :

julia> X = reshape(1:2*3*5*7, 5, 3, 7, 2);

julia> selectdim(X, 3, 7)
5×3×2 view(reshape(::UnitRange{Int64}, 5, 3, 7, 2), :, :, 7, :) with eltype Int64:
[:, :, 1] =
91 96 101
92 97 102
93 98 103
94 99 104
95 100 105

[:, :, 2] =
196 201 206
197 202 207
198 203 208
199 204 209
200 205 210

julia> selectdim(X, 3, 7) == X[:, :, 7, :]
true

关于julia - 如果我不知道总维数,如何仅按一维索引数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55752080/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com