torch 中 [] 中括号的用法 集合

在代码中经常会看见很多[]的用法,这不是函数,让人总是摸不着头脑,现在 mark 一下,方便自己和他人查阅。

a = torch.randn(2,3)
b = a[x, y] 
# x,y = tensor
# 表示 以 x 里的元素为横坐标,y 里的元素为纵坐标,取出 a 中的值到 b

猜你喜欢

转载自blog.csdn.net/qq_40491305/article/details/115180085
今日推荐