Hi everyone,
I just have a general question. Let’s say I have a batch of matrices of shape matrices = [1000, 4, 8]
and I have a batch of scalars scalars = [1000]
, how can I divide the i-th matrix in matrices
with the i-th scalar in scalars
?
I could see a potential way to use torch.stack
but it’d be way too slow as I need the operation to be vectorized! I’m sure I’ve missed something here, any help is apprecitated!