版权属于:
人工智能炼丹师
作品采用:
《
署名-非商业性使用-相同方式共享 4.0 国际 (CC BY-NC-SA 4.0)
》许可协议授权
target.scatter_(dim=1, index=index, src=src)
# 使用 for 循环模拟 scatter_ 操作
for i in range(src.shape[0]):
for j in range(src.shape[1]):
target[i, index[i, j]] = src[i, j]
target.scatter_(dim=0, index=index, src=src)
# 使用 for 循环模拟 scatter_ 操作
for i in range(src.shape[0]):
for j in range(src.shape[1]):
target[index[i, j], j] = src[i, j]
评论 (0)