He

Anh ấy khởi tạo.

Nếu phân phối là TRUNCATED_NORMAL, nó sẽ lấy mẫu từ phân phối chuẩn bị cắt cụt tập trung vào 0 với stddev = sqrt(2 / fanIn) trong đó fanIn là số đơn vị đầu vào trong tensor trọng số.

Nếu phân phối là UNIFORM, nó sẽ lấy các mẫu từ phân phối đồng nhất trong [-limit, limit] , trong đó limit = sqrt(6 / fanIn) (fanIn là số đơn vị đầu vào trong tensor trọng số).

Ví dụ:

Anh bình thường:

     long seed = 1001l;
     He<TFloat32, TFloat32> initializer =
             new org.tensorflow.framework.initializers.He<>(tf,
             Distribution.TRUNCATED_NORMAL, seed););
     Operand<TFloat32> values =
             initializer.call(tf.constant(Shape.of(2,2)), TFloat32.class);
 

Anh ấy đồng phục:

     long seed = 1001l;
     He<TFloat32, TFloat32> initializer =
             new org.tensorflow.framework.initializers.He<>(tf,
             Distribution.UNIFORM, seed););
     Operand<TFloat32> values =
             initializer.call(tf.constant(Shape.of(2,2)), TFloat32.class);
 

GHI CHÚ:

Đối với trình khởi tạo tương đương HeNormal, hãy sử dụng TRUNCATED_NORMAL cho tham số phân phối.

Đối với trình khởi tạo tương đương HeUniform, hãy sử dụng UNIFORM cho tham số phân phối.