擅长:python、mysql、java
<p>您可以使用以下配方:</p>
<p>用分号替换除d外的所有轴,并在d轴上输入值i,例如:</p>
<pre><code>tensorA[0, :] # same as tensorA:select(0,0)
tensorA[:, 1] # same as tensorA:select(1,1)
tensorA[:, 0] # same as tensorA:select(1,0)
</code></pre>
<p>然而,当我尝试这个的时候,我有一个语法错误:</p>
^{pr2}$
<p>所以我用切片代替</p>
<pre><code>i = 1
selection = [slice(0,2,1), i]
tensorA[selection] # same as tensorA:select(1,i)
</code></pre>
<p>此函数的作用是:</p>
<pre><code>def select(t, axis, index):
shape = K.int_shape(t)
selection = [slice(shape[a]) if a != axis else index for a in
range(len(shape))]
return t[selection]
</code></pre>
<p>例如:</p>
<pre><code>import numpy as np
t = K.constant(np.arange(60).reshape(2,5,6))
sub_tensor = select(t, 1, 1)
print(K.eval(sub_tensor)
</code></pre>
<p>印刷品</p>
<blockquote>
<p>[[6., 7., 8., 9., 10., 11.], </p>
<p>[36., 37., 38., 39., 40., 41.]]</p>
</blockquote>