Tensor Calculation Improvement

IoBT-VISTEC / OCTAve

Implementation for OCTAve: 2D en face Optical Coherence Tomography Angiography Vessel Segmentation in Weakly-Supervised Learning with Locality Augmentation (IEEE Transactions on Biomedical Engineering))

https://iobt-vistec.github.io/OCTAve/

11 stars 1 forks source link

Tensor Calculation Improvement #2

Closed p4perf4ce closed 3 years ago

p4perf4ce commented 3 years ago

Proposal

For the sake of readability. Henceforth I declared the prohibition of any Pytorch tensor manipulation method such as

unsqueeze()
squeeze()
stack()
sum()
mean()
min()
etc.

Any of the existing Pytorch tensor manipulation methods that existed previously are subject to be changed.

Discussion

Why?

Because einops and einsum existed. That's all you gonna need.

p4perf4ce commented 3 years ago

Most of operations was replaced with einops. Closing.