The feature_dims option is disabled in get_log_prob, get_entropy, etc.. So you need to manually operate a resulting tensor when you want to keep dimensions other than batch dimension.
Solution
Enabled the feature_dims option.
The dimensional restriction imposed on sum_samples has been lifted.
Issue
The
feature_dimsoption is disabled inget_log_prob,get_entropy, etc.. So you need to manually operate a resulting tensor when you want to keep dimensions other than batch dimension.Solution
feature_dimsoption.