【Swift】CATransform3Dを使って画像を回転したい

以下のサイトをSwift3に書き直してみました、参考にしてみてください。
※UIViewでやっていますが、UIImageViewに置き換えても同じ事はできます。

SwiftでCATransform3Dを使う場合

import UIKit


class ViewController: UIViewController {

    override func viewDidLoad() {
        super.viewDidLoad()

        let view : UIView = UIView(frame: CGRect(x: 100, y: 100, width: 100, height:100))

        view.backgroundColor = UIColor.blue
        view.isUserInteractionEnabled = true
        view.tag = 1
        self.view.addSubview(view)

    }

    override func touchesBegan(_ touches: Set<UITouch>, with event: UIEvent?) {
        super.touchesBegan(touches, with: event)

        if let vi = touches.first?.view {

            switch vi.tag {
            case 1:
                let animation : CABasicAnimation = CABasicAnimation(keyPath: "transform")
                animation.duration = 0.5
                animation.repeatCount = 1
                animation.autoreverses = true

                // X軸
                //let transform : CATransform3D = CATransform3DMakeRotation(CGFloat(M_PI),  1.0, 0.0, 0.0)

                // Y軸
                let transform : CATransform3D = CATransform3DMakeRotation(CGFloat(M_PI),  0.0, 1.0, 0.0)
                animation.toValue = NSValue(caTransform3D : transform)

                vi.layer.add(animation, forKey: "transform")
            default:
                break
            }
        }
    }
}