From b61d0364dbe3ff720f9aedc323e3c7fd40806146 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Carl=20Thom=C3=A9?= Date: Tue, 11 Apr 2017 18:51:21 +0200 Subject: [PATCH] Update README.md --- README.md | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/README.md b/README.md index a92f5e4..b54ca3e 100644 --- a/README.md +++ b/README.md @@ -1,10 +1,8 @@ # TensorFlow ConvLSTM Cell [![Build Status](https://travis-ci.org/carlthome/tensorflow-convlstm-cell.svg?branch=master)](https://travis-ci.org/carlthome/tensorflow-convlstm-cell) A ConvLSTM cell for TensorFlow's RNN API. -# Usage ```python import tensorflow as tf -from cell import ConvLSTMCell batch_size = 32 timesteps = 100 @@ -20,6 +18,7 @@ inputs = tf.placeholder(tf.float32, [batch_size, timesteps] + shape + [channels] inputs = tf.transpose(inputs, (1, 0, 2, 3, 4)) # Add the ConvLSTM step. +from cell import ConvLSTMCell cell = ConvLSTMCell(shape, filters, kernel) outputs, state = tf.nn.dynamic_rnn(cell, inputs, dtype=inputs.dtype, time_major=True)