Как написать функцию потерь, которая должна выполнить некоторые операции с тензорами заполнителя перед запуском сеанса в Tensorflow?

Question

Например, я строю свой AVE-автоматический кодер, но мне нужно сделать некоторые вычисления в слоях кодирования, например взять некоторую строку матрицы W слоев кодирования, чтобы умножить или случайным образом выбрать несколько строк матрицы W. Вот этоsucode:

    # here the X is coding layer parameters in the autoencoder
    def func(X):
        # dosomething in X,such as randomly select some rows to do calulation
        # so i need use the index of row that i want to choose,i let the index
        # to be a placeholder.but it's terminate  before the session begin .
        dosomething
        return