Commit 2ba93df9 by xuchen

update the torch.range function in rpr with the latest API torch.arange

parent d012e8d8
......@@ -319,7 +319,7 @@ class RelativeMultiheadAttention(MultiheadAttention):
range_mat = range_vec.repeat(length, 1)
distance_mat = range_mat - range_mat.transpose(0, 1)
else:
distance_mat = torch.range(-length + 1, 0).view(1, -1)
distance_mat = torch.arange(-length + 1, 1).view(1, -1)
distance_mat_clipped = torch.clamp(distance_mat, -max_relative_length, max_relative_length)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论