Partial

class tensorplay.distributed.tensor.Partial(reduce_op: str = 'sum')[source]

Store values that still need a reduction across one mesh dimension.