Closed
Description
Consider following, np.split is always a list of objects, but after going through Ray, the result is either a single Ray object or a list of Ray objects depending on whether shards
is equal to 1
@ray.method(num_return_vals=shards)
def compute_gradients(self, *weights):
return np.split(array, shards)
Metadata
Metadata
Assignees
Labels
No labels