Fully connected autoencoder with 3D input and output

output = output.view_as(input)