I am working with a PyTorch model (AutoModelForCausalLM) using the transformers library and encountering a RuntimeError related to tensor types and operator support. Here’s a simplified version of my code:

im working on a model that learns chess through ddqn reinforcement learning , in short, in this specific snippet of the code:

I have two tensors in PyTorch: