einops

einops copied to clipboard

Circular imports when importing einops and torch._dynamo

7

**Describe the bug** Importing `einops` before `torch._dynamo` currently leads to warnings. I'm not sure if this needs a fix on the `pytorch` or `einops` side. This is annoying for CI...

befelix

Update README.md

1

fix minor typo

project-delphi

Alias ellipsis to a star for rearrange

3

Hey, guys, the library is great! But, there is a small non-intuitive thing that I consistently stumble upon (I mean, twice is consistent, right?). It is that for rearrange one...

knyazer

Passing a float in `repeat` as a dimension size prevents correct usage afterwards

2

Consider the scenario 0) Working environment is jupyter notebook(that's where this bug affects code in the wild) 1) I call `repeat(image, 'h -> (h H)', H=H)` with `H=2.0` (float, not...

Maykeye

[Feature suggestion] Optionally return the inferred dimensions

3

# Description Einops calculates automatically some of the inferred dimension. For example, in ViT, Einops is great for decomposing a tensor into stacked patches like this: ```python x = torch.ones([1,...

ml-w

issues related to paddle have been fixed in paddle nightly build, update settings here

2

- https://github.com/PaddlePaddle/Paddle/issues/63927 has been fixed by https://github.com/PaddlePaddle/Paddle/pull/67588, so it's possible to use nightly build now. - paddle has established a dedicated pypi index repository, updated the command for install paddlepaddle...

jzhang533

support mindspore

1

lvyufeng

Interactive playground for einops

13

I want to use interactive notebooks as an intro to einops, currently there are two versions: on github, and converted version in docs. Only two notebooks work in jupyterlite (basic...

arogozhnikov

[Feature suggestion] Support of tensordict (and tensorclass)

2

Tensordicts are dicts of tensors with a common batch dimension, tensorclasses dataclasses of tensor. https://pytorch.org/tensordict/ Tensordict seems to be still not fully mature, but is already quite useful. If I...

fzimmermann89

Plan: - [x] move notebooks testing as independent part, that is triggered _only_ in CI - [x] include test suite as a part of pypi distribution - [x] provide a...

arogozhnikov