Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Use num_batch_dims=0 to deal with observations containing scalars #231

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Use num_batch_dims=0 to deal with observations containing scalars #231

wants to merge 1 commit into from

Commits on May 23, 2022

  1. Use num_batch_dims=0 to deal with observations containing scalars 

    When a (nested) observation contain a scalar value on some axes
with dimension 1, batch_concat() will crash due to a shape mismatch
error (e.g., cannot concatenate tensors of shape [100, 1] and []).

This happens because batch_concat by default assumes the input
tensors are batched, but when computing the statistics
the batch dimensions (axis=0) already have been eliminated.
    @wookayin
    wookayin committed May 23, 2022
    Configuration menu
    Copy the full SHA
    6496944 View commit details
    Browse the repository at this point in the history