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

Question

我是 Theano 的新手，我只是想了解一些基本功能。我有一个张量变量 x，我希望函数 return 一个形状相同但填充值为 0.2 的张量变量 y。我不确定如何定义 y。

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

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

masked_array = function([x],y)

Answer 1

可能有十几种不同的方法来执行此操作，哪种方法最好取决于上下文：code/functionality 的这一部分如何适合更广泛的程序。

这是一种方法：

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

How to get a theano function to return the an array of the same length as another tensor variable