提交 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): ...@@ -189,7 +189,7 @@ class RandomStreams(Component):
""" """
return self.gen(raw_random.multinomial, *args, **kwargs) 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""" """Return a variable with every row (rightmost index) shuffled"""
perm = self.permutation(input.ndim-1, input.shape[:-1], input.shape[-1]) perm = self.permutation(input.ndim-1, input.shape[:-1], input.shape[-1])
shuffled = reorder_row_elements(input, perm) shuffled = reorder_row_elements(input, perm)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论