Revision 2c3e80e99f351f45d46fbad158762efbb56889c2 authored by alexggmatthews on 27 June 2016, 09:31:34 UTC, committed by alexggmatthews on 27 June 2016, 09:31:34 UTC
1 parent b808f87
tf_hacks.py
"""
A collection of hacks for tensorflow.
Hopefully we can remove these as the library matures
"""
import tensorflow as tf
def eye(N):
return tf.diag(tf.ones(tf.pack([N, ]), dtype='float64'))
Computing file changes ...
