Returned tensor image without rank

cheind / tf-matplotlib

Seamlessly integrate matplotlib figures as tensorflow summaries.

MIT License

119 stars 17 forks source link

Returned tensor image without rank #4

Open stratomaster31 opened 5 years ago

stratomaster31 commented 5 years ago

Thank you for this repo, the tool is really useful!

My concern is: is it possible that tfmpl returns an image tensor with rank/shape? I want to use the returned tensor of a plot as input for a Conv layer, and using tfmpl would allow to compute the image during training phase, but the returned images have not rank

thx!

cheind commented 4 years ago

sorry for the late response! Currently this feature is not available, but I am happy to accept any PR!