A menudo ves muchos usos de [] en el código. Esta no es una función, lo que confunde a la gente. Márcala ahora para que tú y otros puedan consultarla.
a = torch.randn(2,3)
b = a[x, y]
# x,y = tensor
# 表示 以 x 里的元素为横坐标，y 里的元素为纵坐标，取出 a 中的值到 b
Colección de uso de [] corchetes en antorcha
Supongo que te gusta
Origin blog.csdn.net/qq_40491305/article/details/115180085
Recomendado
Clasificación