原文地址:http://deeplearning.net/software/theano/tutorial/shape_info.html
严格给定Theano变量的形状是不太可能的,当一个特定值被在实时提供到Theano函数是,函数会确定条件的形状。
目前,不管形状Theano的信息在两种方法被使用:
- 当标准形状提前被知道时,为了生成更快的C代码在CPU和GPU的2维卷积。
- 当我们只需要知道形状而不需要具体的值时。
import theano
x = theano.tensor.matrix('x')
f = theano.function([x], (x**2).shape)
theano.printing.debugprint(f)
#MakeVector{dtype='int64'} [id A] '' 2
# |Shape_i{0} [id B] '' 1
# | |x [id C]
# |Shape_i{1} [id D] '' 0
# |x [id C]
输出的编译函数不会包括任意乘法或乘方。Theano移除了他们为了直接计算他们的形状。
Shape Inference Problem 形状确定问题
Theano在图里传播形状的信息。有时候会带来问题。
import numpy
import theano
x = theano.tensor.matrix('x')
y = theano.tensor.matrix('y')
z = theano.tensor.join(0, x, y)
xv = numpy.random.rand(5,4)
yv = numpy.random.rand(3,3)
f = theano.function([x, y], z.shape)
theano.printing.debugprint(f)
#MakeVector{dtype='int64'} [id A] '' 4
# |Elemwise{Add}[(0, 0)] [id B] '' 3
# | |Shape_i{0} [id C] '' 1
# | | |x [id D]
# | |Shape_i{0} [id E] '' 2
# | |y [id F]
# |Shape_i{1} [id G] '' 0
# |x [id D]
f(xv, yv)
# 没有问题
f = theano.function([x,y],z)
theano.orinting.debugprintf(f)
#Join [id A] '' 0
# |TensorConstant{0} [id B]
# |x [id C]
# |y [id D]
f(xv, yv)
# 出错