根据输入矩阵创建大小为n的向量

Question

所以，我在写一个函数。为了开始计算，我需要创建一个向量1。它要乘以输入矩阵a

如果一个矩阵是6x6，它就变成了一个长度为6的向量，由1组成。如果它是4x4，它就变成了一个长度为4的向量，所有的值都在1处。

对于任何给定的矩阵A，我如何使向量的大小合适?

Answer 1

假设您有以下矩阵A:

julia> A=reshape(1:9,3,3)
3×3 reshape(::UnitRange{Int64}, 3, 3) with eltype Int64:
1  4  7
2  5  8
3  6  9

julia> A*ones(size(A,1))
3-element Vector{Float64}:
12.0
15.0
18.0

(如果你指的是1的水平向量，你可以用ones(1,size(A,1))*A)

注意，使用sum函数的dims参数也可以达到同样的效果:

julia> sum(A,dims=2)
3×1 Matrix{Int64}:
12
15
18

或者如果你想要Vector:

julia> vec(sum(A,dims=2))
3-element Vector{Int64}:
12
15
18