вычислить состояние из командных матриц без пространства действия - PullRequest
0 голосов
/ 01 апреля 2020

Я почти уверен, что мои потребности могли быть выражены более четко, но мне не удалось. Я пытаюсь достичь чего-то, и мне интересно, как это сделать. Цель состоит в том, чтобы использовать графический процессор и избегать больших петель ресурсов.

Итак, вот моя проблема. Допустим, у меня есть две команды, позвольте назвать их start_walking и stop_walking. Предполагается, что игрок может на каждом временном шаге вызвать один из этих двух или ничего не делать. Я хочу вычислить результирующее состояние.

Мы представим сработавшую команду как 1, а n oop как 0.

start_walking = torch.tensor([1, 0, 1, 0, 1, 0, 0, 0, 0, 1, 0])
stop_walking  = torch.tensor([0, 0, 0, 0, 0, 0, 1, 0, 1, 0, 0])

# I want, if possible with GPU and as few steps as possible, to obtain:
> is_walking = torch.tensor([1, 1, 1, 1, 1, 1, 0 ,0 ,0 ,1, 1])
...