Revision 1da8382c5b31d43d880e467addee5d950b6972c7 authored by James Hensman on 29 July 2016, 19:38:40 UTC, committed by GitHub on 29 July 2016, 19:38:40 UTC
2 parent s bd0b423 + e155d76
Raw File
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'))
back to top