Skip to content

Add optional batch shape argument to tree_random_like. - #1727

Open
carlosgmartin wants to merge 1 commit into
google-deepmind:mainfrom
carlosgmartin:tree_random_like_shape
Open

Add optional batch shape argument to tree_random_like.#1727
carlosgmartin wants to merge 1 commit into
google-deepmind:mainfrom
carlosgmartin:tree_random_like_shape

Conversation

@carlosgmartin

Copy link
Copy Markdown
Contributor

Adds an optional batch shape argument to tree_random_like.

This avoids having to do a separate call to tree_batch_shape before calling tree_random_like, improving legibility.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant