question [question] 如何用类型安全的方式表达 tensorMap 和 gradient 这类不改变 Tensor 的结构的函数？ 从而避免使用 assertTensorArray 或 assertScalar。