pytorch-summary
pytorch-summary copied to clipboard
support for list as output item, not only tensors
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.