search

googlecolab / colabtools

Python libraries for Google Colaboratory
Apache License 2.0
2.21k stars 725 forks source link

Package request: `einops` #4739

Open Richienb opened 4 months ago

Richienb commented 4 months ago

Is your feature request related to a problem? Please describe.

It's frustrating to keep having to install einops, which I use so often. It's a popular module that lets you rearrange tensors by taking as input the starting shape and ending shape, then automatically applying the necessary transformations.

For example:

from einops import rearrange

output_tensor = rearrange(input_tensor, 't b c -> b c t')

Describe the solution you'd like

Include the einops package by default.

mayankmalik-colab commented 3 months ago

I agree. einops is widely used package. We will try to install by default soon. Tracked internally - b/356851498 . Thanks