How to retrive input values from Flux Dense layer

albheim · October 3, 2022, 6:50am

Seems to be layer.weight to access the weight matrix, where the first dimension of that matrix is the output size and the second dimension is the input size.

github.com

FluxML/Flux.jl/blob/9a8a676e81de4b3b47fd78b9c04b2cf4e16f20c6/src/layers/basic.jl#L153-L161


      
          struct Dense{F, M<:AbstractMatrix, B}
            weight::M
            bias::B
            σ::F
            function Dense(W::M, bias = true, σ::F = identity) where {M<:AbstractMatrix, F}
              b = _create_bias(W, bias, size(W,1))
              new{F,M,typeof(b)}(W, b, σ)
            end
          end

Topic		Replies	Views
Interpretation of the weights returned by Flux in model zoo example New to Julia flux	3	1149	September 21, 2018
Model weights dense General Usage question , flux	5	776	January 10, 2022
Dense Layers, softmax, relu New to Julia	4	2491	March 4, 2020
How can I train a simple model to understand the relation of math addition? [Flux] Machine Learning	5	488	February 7, 2020
Flux.jl, treat rows independently General Usage flux	0	422	September 16, 2019

How to retrive input values from Flux Dense layer

Related topics