使用Tensorflow中的索引切片张量

时间:2017-01-18 09:23:05

标签: arrays tensorflow slice

基本上我有一个2d阵列,我想做这个很好的numpy式的东西

#include <iostream>
using namespace std;

int main(){
    int t;
    long long int n,res,x;
    scanf("%d",&t);
    while(t--){
            scanf("%lld",&n);
            res=0;


    for(int i=0;i<n;i++){
            scanf("%lld",&x);
            res^=x;
    }
    if(res==0)
        printf("-1\n");
    else 
        printf("%lld\n",res);

}
return 0;
}
Tensorflow中的

行和列只是两个整数。

2 个答案:

答案 0 :(得分:14)

事实上,TensorFlow现在可以更好地支持切片,因此您可以使用与NumPy完全相同的语法:

result = noise_spec[:rows, :cols]

答案 1 :(得分:2)

发现了,这是

tf.slice(noise_spec, [0,0],[rows, cols])