Веса niftynet для блока Self-Attention - PullRequest
0 голосов
/ 25 октября 2018

есть ли способ получить вес свертки во время тренировки с помощью класса ConvolutionLayer?

На самом деле я пытаюсь реализовать «блок» самоуважения в niftynet (уровень самообслуживания («Генерирующие противоборствующие сети самообслуживания» (SAGAN), Чжан и др. https://arxiv.org/pdf/1805.08318.pdf), нодля этого нам нужны функции изображения.

Я видел, что это можно сделать с помощью тензорного потока, используя:

def get_weights_variable(layer_name):
     with tf.variable_scope(layer_name,reuse=True):
           weights=tf.get_variable('kernel')
      return weights

, так что это эквивалент для niftynet?

Если нет никакой идеи о том, как мы могли бы реализовать блок самоконтроля?

С наилучшими пожеланиями

Пол

...