layers.py 文件源码

def get_plots_out(self, sample=0, frames=slice(0, None)):
        """Prepare to plot outputs for sample 'sample' with 'frames' frames"""
        plotsink = list()
        plot_dict = dict()
        plot_range_dict = dict()

        # Prepare output for plotting (plot_dict value is [tensor, [min, max]]
        plot_dict['{}_out'.format(self.name)] = tf.arg_max(self.out[sample, frames, :, :, :], 3)
        plot_range_dict['{}_out'.format(self.name)] = [0, self.n_units]
        plotsink.append(['{}_out'.format(self.name)])

        return plot_dict, plotsink, plot_range_dict