/
githubmirror
/
the-algorithm
Обзор
Документация
Войти
/
githubmirror
/
the-algorithm
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
twml/twml_common/initializer.py
14 строк
477 B
twitter-team
Twitter Recommendation Algorithm
01 апр 2023, 01:36
01 апр 2023, 01:36
ef4c5eb
Код
Авторство
О чём код?
import tensorflow.compat.v1 as tf class PartitionInitializer(tf.keras.initializers.Initializer): """Required to initialize partitioned weight with numpy array for tests""" def __init__(self, np_array): self.np_array = np_array def __call__(self, shape, dtype=None, partition_info=None): offset = partition_info.var_offset ix0, ix1 = offset[0], offset[0] + shape[0] iy0, iy1 = offset[1], offset[1] + shape[1] return self.np_array[ix0:ix1, iy0:iy1]