нарезка и индексирование в Теано

Question

Я хочу, чтобы индексировать переменную Tensor в Теано, как, например:

• х имеет тип theano.tensor.var.TensorVariable (например [[1,2,3],[4,5,6],[7,8,9]])

Я хочу, чтобы получить [[1,2],[4,5],[7,8]], и [[2,3],[5,6],[8,9]].

Для numpy varaible я бы просто сделал x[:,0:-1] и x[:,1:x.shape[0]] соответственно, но я не могу понять, как получить результаты, которые я хочу в Theano.

Answer 1

Вы бы сделать это так же, как в Теано, как вы делаете в NumPy:

import theano 
import theano.tensor as T 

x = T.imatrix('x') 
y = x[:, 0: -1] 
z = x[:, 1: x.shape[0]] 

f = theano.function([x], y) 
g = theano.function([x], z) 

x_ = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 
print(f(x_)) 
print(g(x_))