search

synnada-ai / mithril

Mithril: A Modular Machine Learning Library for Model Composability
Apache License 2.0
31 stars 8 forks source link

[BUG] Tensor slicing does not work with ellipsis #38

Closed aturker-synnada closed 6 days ago

aturker-synnada commented 6 days ago

Describe the Bug

Tensor item does not work with ellipsis slice input.

To Reproduce

input = IOKey("input", shape=(3, 4, 5))
model = Model()
model += Buffer()(input=input[..., 3], output="output")

Expected Behavior

Should work as Numpy tensor slicing.

System Info