提交 cadbdfb9 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

merge

差异被折叠。
......@@ -6,7 +6,7 @@ import numpy
from theano.compile import module, In, Component
from theano.gof import Container
from theano.tensor import raw_random, reorder_row_elements
from theano.tensor import raw_random, permute_row_elements
class RandomStreamsInstance(object):
"""RandomStreamsInstance"""
......@@ -192,7 +192,7 @@ class RandomStreams(Component):
def shuffle_row_elements(self, input):
"""Return a variable with every row (rightmost index) shuffled"""
perm = self.permutation(input.ndim-1, input.shape[:-1], input.shape[-1])
shuffled = reorder_row_elements(input, perm)
shuffled = permute_row_elements(input, perm)
return shuffled
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论