search

BiomedSciAI / fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)
Apache License 2.0
134 stars 34 forks source link

min_size_per_dim + changes for scalars inputs+outputs support #354

Closed YoelShoshan closed 2 months ago

YoelShoshan commented 4 months ago

by allowing min_size_per_dim to be longer than the number of dims in a tensor, it makes usage easier.

for example,

min_size_per_dim=(10_000, -1) would then work for tensors of single dim - e.g. of size (1234, ) and also for tensors of 2(+) dims - e.g. of size (4321, 100)

this is much more nice for the user in several cases.

also - few changes for scalars inputs+outputs