CODEWITHSUNDEEP
pythonluakerasconv-neural-networktorch
shohkhan
shohkhan

Reputation: 85

Keras function like View in Torch

I am working on converting some code in Torch and lua to Keras and python. Is there a function in Keras that works like View in Torch? EDIT: I want to use the function to convert the output from a convolutional layer to proper shape as a input to a fully connected layer.

Upvotes: 1

Views: 304

Answers (1)

shohkhan
shohkhan

Reputation: 85

I actually got the answer. I can use Reshape. However, for my case I am using Flatten to "flatten" the the shape before passing to a fully connected layer.

Upvotes: 2

Related Questions