提交 63471cda authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Rename function to be more consistent with op name

上级 a3ae249e
......@@ -189,7 +189,7 @@ class RandomStreams(Component):
"""
return self.gen(raw_random.multinomial, *args, **kwargs)
def shuffle_rows(self, input):
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)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论