Hello, I am trying to create an array a with shape [T, T, T, 3], with T = 2 or 3

array[T - 1, 0, 0] = RGB color A

array[0, T - 1, 0] = RGB color B

array[0, 0, T - 1] = RGB color C

with other elements to be the linear combinations of A, B, C.

array[0, 0, 0] is 0

array[T - 1, T - 1, T - 1] = A + B + C

array[T / 3, T / 3, T / 3] = A / 3 + B / 3 + C / 3 (with T = 3)

array[T / 2, T / 2, 0] = A / 2 + B / 2 (with T = 2)

etc.

Now I suspect what I want to do can be done with torch.meshgrid(T,T), however, I don’t seem to be able to do it.

Any help will be appreciated!