Revision 7172bd3497ed97f67a35c85d03675f7abc907044 authored by James Hensman on 10 July 2016, 07:01:58 UTC, committed by James Hensman on 10 July 2016, 07:01:58 UTC
1 parent dc5ac86
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 ...
