Trying out more tensor math operations