Method zero

codegeex/megatron/model/distributed.py:37–39

Reset the buffer to zero.

(self)

        )

    def zero(self):
        """Reset the buffer to zero."""
        self.data.zero_()

    def get(self, shape, start_index):
        """Return a tensor with the input `shape` as a view into the