keep_constraint

Contents

keep_constraint#

class brainpy.math.sharding.keep_constraint(x)[source]#

Keep the sharding constraint of the given inputs during computation.

Parameters:

x (Any) – Any.

Returns:

Same as x.

Return type:

constraint_x