Skip to content

Instantly share code, notes, and snippets.

@benjaminkaplanphd

benjaminkaplanphd/theano_shape.py

Created September 13, 2019 15:12
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save benjaminkaplanphd/17d0161fd2da1ded8f12d8df64cbc971 to your computer and use it in GitHub Desktop.
Save benjaminkaplanphd/17d0161fd2da1ded8f12d8df64cbc971 to your computer and use it in GitHub Desktop.
Download ZIP
Print Theano Tensor Shape
Raw
theano_shape.py
import numpy as np
from theano import function, printing, shared, tensor as T
PRINT = printing.Print('Theano shape: ', attrs=['shape'])
def example():
array = np.ones((3, 4))
print(f'Numpy shape: {array.shape}')
x = T.dmatrix()
printed_x = PRINT(x)
f = function([x], printed_x)
f(array)
example()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment