pytorch-summary icon indicating copy to clipboard operation
pytorch-summary copied to clipboard

support for list as output item, not only tensors

Open leora-betesh opened this issue 4 years ago • 0 comments

Add support for a net outputting a list. Say you are returning intermediary feature-maps from a U-net. These feature-maps will have a different size and cannot be stacked into one tensor. Instead they can be placed in a list.

leora-betesh avatar Mar 11 '21 12:03 leora-betesh