50
правок
Изменения
tensorflow he
conv1 = torch.nn.Conv2d(...)
torch.nn.init.kaiming_uniform_(conv1.weight)
====Пример инициализации He на языке Python с библиотекой TensorFlow====
<font color=darkgreen># инициализация параметров тензора 2x2</font>
initializer = tf.keras.initializers.HeUniform()
values = initializer(shape=(2, 2))
==См.также==
