theano.tensor.split looks very similar. On Tue, Nov 29, 2016 at 12:44 AM, Ragav Venkatesan < [email protected]> wrote:
> 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. > -- --- 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.
