Fix torch rules (#581)

This commit is contained in:
DS 2022-01-30 01:23:05 +01:00 committed by GitHub
parent ecea0a2896
commit 4eea0834b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -27,9 +27,13 @@ local torch_get_output_rules = function(node)
return rotate_torch_rules(rules, node.param2) return rotate_torch_rules(rules, node.param2)
end end
local torch_input_rules_unrotated_horizontal = {vector.new(-2, 0, 0), vector.new(-1, 1, 0)}
local torch_input_rules_unrotated_vertical = {vector.new(-2, 0, 0)}
local torch_get_input_rules = function(node) local torch_get_input_rules = function(node)
local rules = {{x = -2, y = 0, z = 0}, local rules = (node.param2 == 0 or node.param2 == 1)
{x = -1, y = 1, z = 0}} and torch_input_rules_unrotated_vertical
or torch_input_rules_unrotated_horizontal
return rotate_torch_rules(rules, node.param2) return rotate_torch_rules(rules, node.param2)
end end