I think you can write your own function that will do this by using the 
set_tensor method. 

On Monday, November 28, 2016 at 4:32:46 PM UTC-7, A M wrote:
>
> Does Theano support tensor split similar to the following in TensorFlow?
>
> tf.split(split_dim, num_split, value, name='split')
>
> Splits a tensor into num_split tensors along one dimension.
>
> Splits value along dimension split_dim into num_split smaller tensors. 
> Requires that num_split evenly divide value.shape[split_dim].
>
> For example:
>
> # 'value' is a tensor with shape [5, 30]# Split 'value' into 3 tensors along 
> dimension 1
> split0, split1, split2 = tf.split(1, 3, value)
> tf.shape(split0) ==> [5, 10]
>
>

-- 

--- 
You received this message because you are subscribed to the Google Groups 
"theano-users" group.
To unsubscribe from this group and stop receiving emails from it, send an email 
to [email protected].
For more options, visit https://groups.google.com/d/optout.

Reply via email to