2016-10-28 3 views
1

인덱싱 나는 같은 인덱스에 Theano에서 텐서 변수를 원하는 :슬라이스와 Theano

  • X가 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의 경우, 나는 단지 x[:,0:-1]x[:,1:x.shape[0]]을 각각 쓰지만, 나는 Theano에서 원하는 결과를 얻는 방법을 알 수 없다. 당신이 NumPy와에서하고있는 것처럼

답변

2

당신은 그것을 Theano에서 같은 방법으로 할 것이다 :

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_))