如何获得一个theano函数来返回与另一个张量变量长度相同的数组

Question

我是Theano的新手，我只想弄清楚一些基本功能。我有张量变量x，我希望函数返回相同形状的张量变量y，但填充值0.2。我不知道如何定义y。

例如，如果x = [1,2,3,4,5]，那么我想y = [0,2,0,2,0,2,0,2,0]

from theano import tensor, function
y = tensor.dmatrix('y')

masked_array = function([x],y)

Answer 1

可能有十几种不同的方法可以做到这一点，哪种方式最好取决于上下文：这段代码/功能如何适用于更广泛的程序。

这是一种方法：

import theano
import theano.tensor as tt

x = tt.vector()
y = tt.ones_like(x) * 0.2
f = theano.function([x], outputs=y)
print f([1, 2, 3, 4, 5])