예제 #1
0
 def typ_apply(self, inputs):
     return core.TensorType(inputs[0].dtype, 4)
예제 #2
0
 def typ_apply(self, inputs):
     return core.TupleType(core.TensorType(inputs[0].dtype, 4),
                           core.TensorType('i4', 4))
예제 #3
0
 def typ_apply(self, inputs):
     assert inputs[0].ndim == 4
     return core.TensorType(inputs[0].dtype, 4)
예제 #4
0
 def typ_apply(self, inputs):
     assert inputs[0].dtype == 'f4'
     d = inputs[0].ndim
     return core.TupleType(core.TensorType(cgt.floatX, d),
                           core.TensorType(cgt.floatX, d))
예제 #5
0
 def typ_apply(self, inputs):
     ndim = inputs[0].ndim
     return core.TupleType(core.TensorType(cgt.floatX, ndim),
                           core.TensorType(cgt.floatX, ndim))
예제 #6
0
 def typ_apply(self, _inputs):
     return core.TensorType(cgt.floatX, 4)