Obter coluna diferente em cada linha
Eu gostaria de obter uma coluna diferente de cada linha de uma matriz. Por exemplo:
A = [1,2;1,4;5,2]
B = [2;2;1]
a saída deve render:
out = [2;4;5]
Então, em suma:A
é a matriz eB
tem os índices para as colunas por linha. Como posso fazer isso sem usar um loop (se for possível)?