Tensor Shape and Attributes

Problem Statement

Given a PyTorch tensor, extract and return its key attributes.

Background

Understanding tensor attributes is crucial for debugging and ensuring tensors are compatible for operations. Every tensor has a shape, data type, and device.

Your Task

Write a function get_tensor_info(tensor) that returns a dictionary containing the tensor's shape (as a list), dtype (as a string), and device (as a string).

Output Format

Return a dictionary with keys: "shape", "dtype", "device".