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 114e439 commit cbf55abCopy full SHA for cbf55ab
pymc/data.py
@@ -321,7 +321,7 @@ def __init__(
321
else:
322
data = np.asarray(data, dtype)
323
in_memory_slc = self.make_static_slices(in_memory_size)
324
- self.shared = aesara.shared(data[in_memory_slc])
+ self.shared = aesara.shared(data[tuple(in_memory_slc)])
325
self.update_shared_f = update_shared_f
326
self.random_slc = self.make_random_slices(self.shared.shape, batch_size, random_seed)
327
minibatch = self.shared[self.random_slc]
0 commit comments