We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 630bb27 commit c61846dCopy full SHA for c61846d
xbatcher/generators.py
@@ -53,6 +53,7 @@ def _drop_input_dims(ds, input_dims, suffix='_input'):
53
out.coords[dim] = newdim, ds[dim].data, ds[dim].attrs
54
return out
55
56
+
57
def _maybe_stack_batch_dims(
58
ds, input_dims, squeeze_batch_dim, stacked_dim_name='sample'
59
):
0 commit comments