base.py 文件源码

def train_op(self) -> tf.Operation:
        """
        Returns an operation for performing a single training step.

        Returns
        -------
        tf.Operation
            An operation for performing a single training step
        """
        return self.graph.get_collection("train_op")[0]